test: Update for F90 datatype constructor returning the NULL handle

mpi4py · Dec 15, 2023 · c8735ed · c8735ed
1 parent a09fc75
commit c8735ed
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/test_datatype.py b/test/test_datatype.py
@@ -505,6 +505,8 @@ def check(self, oldtype, factory, *args):
                 newtype = factory(oldtype, *args)
             else:
                 newtype = factory(*args)
+            if newtype == MPI.DATATYPE_NULL or newtype.Get_size() == 0:
+                raise NotImplementedError
         except NotImplementedError:
             self.skipTest('mpi-type-constructor')
         self.check_contents(factory, newtype, oldtype)
@@ -524,6 +526,8 @@ def check(self, oldtype, factory, *args):
                 newtype0 = factory(oldtype, *args)
             else:
                 newtype0 = factory(*args)
+            if newtype0 == MPI.DATATYPE_NULL or newtype0.Get_size() == 0:
+                raise NotImplementedError
         except NotImplementedError:
             self.skipTest('mpi-type-constructor')
         newtype1 = loads(dumps(newtype0))