update test for nan, inf, ninf

NeurodataWithoutBorders · Mar 20, 2024 · 8567604 · 8567604
1 parent 49d023b
commit 8567604
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -252,9 +252,14 @@ def test_nan_inf_attributes():
             X2 = client["X"]
             assert isinstance(X2, lindi.LindiH5pyDataset)
 
-            assert X2.attrs["nan"] == "NaN"
-            assert X2.attrs["inf"] == "Infinity"
-            assert X2.attrs["ninf"] == "-Infinity"
+            nanval = X1.attrs["nan"]
+            assert isinstance(nanval, float) and np.isnan(nanval)
+            assert X1.attrs["inf"] == np.inf
+            assert X1.attrs["ninf"] == -np.inf
+
+            # assert X2.attrs["nan"] == "NaN"
+            # assert X2.attrs["inf"] == "Infinity"
+            # assert X2.attrs["ninf"] == "-Infinity"
 
 
 def test_reference_file_system_to_file():