Attempt to catch unit test failures more reliably

tskisner · Mar 14, 2024 · 88d4513 · 88d4513
1 parent 7dbb4ed
commit 88d4513
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/pshmem/test.py b/pshmem/test.py
@@ -471,5 +471,17 @@ def run():
     suite.addTest(unittest.makeSuite(LockTest))
     suite.addTest(unittest.makeSuite(ShmemTest))
     runner = unittest.TextTestRunner()
-    runner.run(suite)
+
+    ret = 0
+    _ret = runner.run(suite)
+    if not _ret.wasSuccessful():
+        ret += 1
+
+    if self.comm is not None:
+        ret = self.comm.allreduce(ret, op=MPI.SUM)
+
+    if ret > 0:
+        print(f"{ret} Processes had failures")
+        sys.exit(6)
+
     return