TST: Compare nCell against array size

rochefort-lab · Jul 9, 2021 · 934beb0 · 934beb0
1 parent 148db90
commit 934beb0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fissa/tests/test_core.py b/fissa/tests/test_core.py
@@ -79,8 +79,8 @@ def compare_output(self, actual, separated=True, compare_deltaf=None):
         self.assert_allclose_ragged(actual.roi_polys, self.expected["roi_polys"])
         # Check parameters match
         self.assert_equal(actual.expansion, self.expected["expansion"])
-        self.assert_equal(actual.nCell, self.expected["nCell"])
         self.assert_equal(actual.nRegions, self.expected["nRegions"])
+        self.assert_equal(actual.nCell, len(actual.raw))
         if separated:
             # Check sizes are correct
             self.assert_equal(np.shape(actual.sep), expected_shape)