distinguish datatypes in test assertions

brainglobe · Apr 29, 2024 · 8b8d6e1 · 8b8d6e1
1 parent 69988fa
commit 8b8d6e1
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/tests/tests/test_unit/test_regions/test_region_analysis.py b/tests/tests/test_unit/test_regions/test_region_analysis.py
@@ -99,7 +99,10 @@ def test_summarise_brain_regions_all_not_empty(
     df = pd.read_csv(filename)
     assert df.shape == (3, 11)
     for key, value in previous_world_space_values.items():
-        assert np.isclose(df[key].iloc[0],value)
+        if isinstance(value, (int, float)):
+            assert np.isclose(df[key].iloc[0], value)
+        else:
+            assert df[key].iloc[0] == value
 
 
 def test_summarise_brain_regions_some_empty(
@@ -124,8 +127,10 @@ def test_summarise_brain_regions_some_empty(
     df = pd.read_csv(filename)
     assert df.shape == (1, 11)
     for key, value in previous_world_space_values.items():
-        assert np.isclose(df[key].iloc[0], value)
-
+        if isinstance(value, (int, float)):
+            assert np.isclose(df[key].iloc[0], value)
+        else:
+            assert df[key].iloc[0] == value
 
 def test_summarise_brain_regions_all_empty(
     labels_layer, empty_labels_layer, tmp_path