fixup

Author: PierreMarchand20

.github/workflows/CI.yml

@@ -135,25 +135,18 @@ jobs:
       - name: Run tests
         run: |
           export PYTHONPATH="$PWD/build"
-          export GCOV_PREFIX_STRIP=2
-          COVDIR=/tmp/gcov-mpi
-          mkdir -p $COVDIR
-
-          # Wrap mpirun calls so each rank writes to its own subdir
-          run_mpi_pytest() {
-            local np=$1
-            shift
-            mpirun -np $np ${{ matrix.MPIEXEC_PREFLAGS }} \
-              bash -c 'export GCOV_PREFIX='"$COVDIR"'/${OMPI_COMM_WORLD_RANK}; \
-                       mkdir -p $GCOV_PREFIX; \
-                       python3 -m pytest "$@"' bash "$@"
-          }
-          
-          run_mpi_pytest 1 tests --ignore=tests/test_hmatrix.py --ignore=tests/test_logger.py
-          run_mpi_pytest 2 tests --ignore=tests/test_hmatrix.py --ignore=tests/test_logger.py
-          run_mpi_pytest 3 tests --ignore=tests/test_hmatrix.py --ignore=tests/test_logger.py
-          run_mpi_pytest 4 tests --ignore=tests/test_hmatrix.py --ignore=tests/test_logger.py
-
+          mpirun -np 1 python3 -m pytest tests/test_cluster
+          mpirun -np 2 python3 -m pytest tests/test_cluster
+          mpirun -np 3 python3 -m pytest tests/test_cluster
+          mpirun -np 4 python3 -m pytest tests/test_cluster
+          mpirun -np 1 python3 -m pytest tests/test_distributed_operator
+          mpirun -np 2 python3 -m pytest tests/test_distributed_operator
+          mpirun -np 3 python3 -m pytest tests/test_distributed_operator
+          mpirun -np 4 python3 -m pytest tests/test_distributed_operator
+          mpirun -np 1 python3 -m pytest tests/test_ddm_solver
+          mpirun -np 2 python3 -m pytest tests/test_ddm_solver
+          mpirun -np 3 python3 -m pytest tests/test_ddm_solver
+          mpirun -np 4 python3 -m pytest tests/test_ddm_solver
           python3 -m pytest tests/test_hmatrix.py
           python3 -m pytest tests/test_logger.py
 
@@ -179,7 +172,6 @@ jobs:
                --capture \
                --base-directory ./ \
                --directory build \
-               --directory $COVDIR \
                --output-file coverage.info
 
           # Exclude third-party and system paths

tests/test_distributed_operator.py

@@ -96,6 +96,11 @@ def test_distributed_operator(
     Y_2 = generator.mat_mat(X)
     assert np.linalg.norm(Y_1 - Y_2) / np.linalg.norm(Y_2) < epsilon
 
+    X = np.asfortranarray(np.random.rand(nb_cols, 1))
+    Y_1 = distributed_operator @ X
+    Y_2 = generator.mat_mat(X)
+    assert np.linalg.norm(Y_1 - Y_2) / np.linalg.norm(Y_2) < epsilon
+
     # Test sub matrix vector product
     test_offset = int(nb_cols / 10)
     test_size = int(nb_cols / 10)