Improve MPI unit tests runner

hpc4cmb · Jan 1, 2025 · 6eef58a · 6eef58a
1 parent c54f249
commit 6eef58a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -109,5 +109,5 @@ jobs:
             && export OMP_NUM_THREADS=1 \
             && mkdir -p test \
             && pushd test >/dev/null 2>&1 \
-            && mpirun -np 2 python -c 'import flacarray.tests; flacarray.tests.run()' \
+            && mpirun -np 2 python -m mpi4py -c 'import flacarray.tests; flacarray.tests.run()' \
             && popd >/dev/null 2>&1
diff --git a/flacarray/tests/runner.py b/flacarray/tests/runner.py
@@ -5,6 +5,8 @@
 import sys
 import unittest
 
+from ..mpi import use_mpi, MPI
+
 from . import array as test_array
 from . import bindings as test_bindings
 from . import hdf5 as test_hdf5
@@ -41,6 +43,8 @@ def test(name=None):
     _ret = runner.run(suite)
     if not _ret.wasSuccessful():
         ret = 1
+        if use_mpi:
+            MPI.COMM_WORLD.Abort(6)
 
     if ret > 0:
         print("Some tests failed", flush=True)