Merge pull request #248 from GeoStat-Framework/float-comparison

Change float comparison in tests
GeoStat-Framework · Aug 9, 2022 · dda2431 · dda2431
2 parents 51204f4 + 0bcb322
commit dda2431
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/tests/test_transform.py b/tests/test_transform.py
@@ -63,19 +63,19 @@ def test_transform_normal(self):
         srf.transform(
             "discrete", values=values, thresholds=thresholds, store="f1"
         )
-        np.testing.assert_array_equal(np.unique(srf.f1), [-1, 0, 1])
+        np.testing.assert_allclose(np.unique(srf.f1), [-1, 0, 1])
 
         values = [-1, 0, 1]
         srf.transform(
             "discrete", values=values, thresholds="arithmetic", store="f2"
         )
-        np.testing.assert_array_equal(np.unique(srf.f2), [-1.0, 0.0, 1.0])
+        np.testing.assert_allclose(np.unique(srf.f2), [-1.0, 0.0, 1.0])
 
         values = [-1, 0, 0.5, 1]
         srf.transform(
             "discrete", values=values, thresholds="equal", store="f3"
         )
-        np.testing.assert_array_equal(np.unique(srf.f3), values)
+        np.testing.assert_allclose(np.unique(srf.f3), values)
         # checks
         with self.assertRaises(ValueError):
             srf.transform("discrete", values=values, thresholds=[1])
@@ -84,7 +84,7 @@ def test_transform_normal(self):
 
         # function
         srf.transform("function", function=lambda x: 2 * x, store="f4")
-        np.testing.assert_array_equal(2 * srf.field, srf.f4)
+        np.testing.assert_allclose(2 * srf.field, srf.f4)
         with self.assertRaises(ValueError):
             srf.transform("function", function=None)
 
@@ -162,7 +162,7 @@ def test_transform_denormal(self):
             store="f1",
             process=True,
         )
-        np.testing.assert_array_equal(np.unique(np.log(srf.f1)), [-1, 0, 1])
+        np.testing.assert_allclose(np.unique(np.log(srf.f1)), [-1, 0, 1])
 
         values = [-1, 0, 1]
         srf.transform(
@@ -172,9 +172,7 @@ def test_transform_denormal(self):
             store="f2",
             process=True,
         )
-        np.testing.assert_array_equal(
-            np.unique(np.log(srf.f2)), [-1.0, 0.0, 1.0]
-        )
+        np.testing.assert_allclose(np.unique(np.log(srf.f2)), [-1.0, 0.0, 1.0])
 
         values = [-1, 0, 0.5, 1]
         srf.transform(
@@ -184,7 +182,7 @@ def test_transform_denormal(self):
             store="f3",
             process=True,
         )
-        np.testing.assert_array_equal(np.unique(np.log(srf.f3)), values)
+        np.testing.assert_allclose(np.unique(np.log(srf.f3)), values)
 
 
 if __name__ == "__main__":