CI: add single mpi sparse test

devitocodes · Jul 1, 2024 · daa9ee7 · daa9ee7
1 parent 1148087
commit daa9ee7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_data.py b/tests/test_data.py
@@ -208,6 +208,17 @@ def test_indexing_into_sparse(self):
         sf.data[1:-1, 0] = np.arange(8)
         assert np.all(sf.data[1:-1, 0] == np.arange(8))
 
+    @pytest.mark.parallel(mode=1)
+    def test_indexing_into_sparse_subfunc_singlempi(self, mode):
+        grid = Grid(shape=(4, 4))
+        s = SparseFunction(name='sf', grid=grid, npoint=1)
+        coords = np.random.rand(*s.coordinates.data.shape)
+        s.coordinates.data[:] = coords
+
+        s.coordinates.data[-1, :] = s.coordinates.data[-1, :] / 2
+
+        assert np.allclose(s.coordinates.data[-1, :], coords[-1, :] / 2)
+
 
 class TestLocDataIDX:
     """