MPI-4.1: Add lowercase Request.get_status_{any|all|some}

src/mpi4py/MPI.pyi

@@ -455,13 +455,19 @@ class Request:
     @classmethod
     def testany(cls, requests: Sequence[Request], status: Status | None = None) -> tuple[int, bool, Any | None]: ...
     @classmethod
+    def get_status_any(cls, requests: Sequence[Request], status: Status | None = None) -> tuple[int, bool]: ...
+    @classmethod
     def waitall(cls, requests: Sequence[Request], statuses: list[Status] | None = None) -> list[Any]: ...
     @classmethod
     def testall(cls, requests: Sequence[Request], statuses: list[Status] | None = None) -> tuple[bool, list[Any] | None]: ...
     @classmethod
+    def get_status_all(cls, requests: Sequence[Request], statuses: list[Status] | None = None) -> bool: ...
+    @classmethod
     def waitsome(cls, requests: Sequence[Request], statuses: list[Status] | None = None) -> tuple[list[int] | None, list[Any] | None]: ...
     @classmethod
     def testsome(cls, requests: Sequence[Request], statuses: list[Status] | None = None) -> tuple[list[int] | None, list[Any] | None]: ...
+    @classmethod
+    def get_status_some(cls, requests: Sequence[Request], statuses: list[Status] | None = None) -> list[int] | None: ...
     def cancel(self) -> None: ...
     handle: int
 

src/mpi4py/MPI/Request.pyx

@@ -357,6 +357,17 @@ cdef class Request:
         return (index, <bint>flag, msg)
     #
     @classmethod
+    def get_status_any(
+        cls,
+        requests: Sequence[Request],
+        Status status: Status | None = None,
+    ) -> tuple[int, bool]:
+        """
+        Non-destructive test for the completion of any requests.
+        """
+        return Request.Get_status_any(requests, status)
+    #
+    @classmethod
     def waitall(
         cls,
         requests: Sequence[Request],
@@ -382,6 +393,17 @@ cdef class Request:
         return (<bint>flag, msg)
     #
     @classmethod
+    def get_status_all(
+        cls,
+        requests: Sequence[Request],
+        statuses: list[Status] | None = None,
+    ) -> bool:
+        """
+        Non-destructive test for the completion of all requests.
+        """
+        return Request.Get_status_all(requests, statuses)
+    #
+    @classmethod
     def waitsome(
         cls,
         requests: Sequence[Request],
@@ -403,6 +425,17 @@ cdef class Request:
         """
         return PyMPI_testsome(requests, statuses)
     #
+    @classmethod
+    def get_status_some(
+        cls,
+        requests: Sequence[Request],
+        statuses: list[Status] | None = None,
+    ) -> list[int] | None:
+        """
+        Non-destructive test for completion of some requests.
+        """
+        return Request.Get_status_some(requests, statuses)
+    #
     def cancel(self) -> None:
         """
         Cancel a request.

src/mpi4py/util/pkl5.py

@@ -564,6 +564,16 @@ def wait(self, status=None):
         """Wait for a request to complete."""
         return _test(self, MPI.Request.Waitall, status)[1]
 
+    @classmethod
+    def get_status_all(cls, requests, statuses=None):
+        """Non-destructive test for the completion of all requests."""
+        arglist = [requests]
+        if statuses is not None:
+            ns, nr = len(statuses), len(requests)
+            statuses += [Status() for _ in range(ns, nr)]
+            arglist.append(statuses)
+        return all(map(Request.get_status, *arglist))
+
     @classmethod
     def testall(cls, requests, statuses=None):
         """Test for the completion of all requests."""

src/mpi4py/util/pkl5.pyi

@@ -52,6 +52,12 @@ class Request(tuple[MPI.Request, ...]):
         status: Status | None = None,
     ) -> Any: ...
     @classmethod
+    def get_status_all(
+        cls,
+        requests: Sequence[Request],
+        statuses: list[Status] | None = None,
+    ) -> bool: ...
+    @classmethod
     def testall(
         cls,
         requests: Sequence[Request],
@@ -61,7 +67,7 @@ class Request(tuple[MPI.Request, ...]):
     def waitall(
         cls,
         requests: Sequence[Request],
-            statuses: list[Status] | None = None,
+        statuses: list[Status] | None = None,
     ) -> list[Any]: ...
 
 class Message(tuple[MPI.Message, ...]):

test/test_request.py

@@ -97,6 +97,10 @@ def testGetStatusAny(self):
             self.assertEqual(status.source, MPI.ANY_SOURCE)
             self.assertEqual(status.tag, MPI.ANY_TAG)
             self.assertEqual(status.error, MPI.SUCCESS)
+        with self.catchNotImplementedError(4, 1):
+            index, flag = MPI.Request.get_status_any(self.REQUESTS)
+            self.assertEqual(index, MPI.UNDEFINED)
+            self.assertTrue(flag)
 
     def testWaitall(self):
         MPI.Request.Waitall(self.REQUESTS)
@@ -139,6 +143,9 @@ def testGetStatusAll(self):
                 self.assertEqual(status.source, MPI.ANY_SOURCE)
                 self.assertEqual(status.tag, MPI.ANY_TAG)
                 self.assertEqual(status.error, MPI.SUCCESS)
+        with self.catchNotImplementedError(4, 1):
+            flag = MPI.Request.get_status_all(self.REQUESTS)
+            self.assertTrue(flag)
 
     def testWaitsome(self):
         ret = MPI.Request.Waitsome(self.REQUESTS)
@@ -171,6 +178,9 @@ def testGetStatusSome(self):
             self.assertIsNone(indices)
             indices = MPI.Request.Get_status_some(self.REQUESTS, statuses)
             self.assertIsNone(indices)
+        with self.catchNotImplementedError(4, 1):
+            indices = MPI.Request.get_status_some(self.REQUESTS)
+            self.assertIsNone(indices)
 
 
 if __name__ == '__main__':

test/test_util_pkl5.py

@@ -239,6 +239,33 @@ def testISSendCancel(self):
                 req.Free()
             self.assertFalse(req)
 
+    def testGetStatusAll(self):
+        comm = self.COMM
+        size = comm.Get_size()
+        rank = comm.Get_rank()
+        requests = []
+        for smess in messages:
+            req = comm.issend(smess, rank)
+            requests.append(req)
+        with self.catchNotImplementedError(4, 1):
+            flag = self.RequestType.get_status_all(requests)
+            self.assertFalse(flag)
+        comm.barrier()
+        for smess in messages:
+            rmess = comm.recv(None, rank)
+            self.assertEqual(rmess, smess)
+        with self.catchNotImplementedError(4, 1):
+            flag = False
+            statuses = []
+            while not flag:
+                flag = self.RequestType.get_status_all(requests, statuses)
+            self.assertEqual(len(statuses), len(requests))
+            for status in statuses:
+                self.assertIsInstance(status, MPI.Status)
+        flag, obj = self.RequestType.testall(requests)
+        self.assertTrue(flag)
+        self.assertEqual(obj, [None]*len(messages))
+
     def testTestAll(self):
         comm = self.COMM
         size = comm.Get_size()