Merge pull request #1113 from wright-group/randint_tests

refine numeric tests
wright-group · Jan 16, 2023 · 3af21a7 · 3af21a7
2 parents 3d51eb3 + 9994538
commit 3af21a7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/dataset/ipow.py b/tests/dataset/ipow.py
@@ -61,7 +61,7 @@ def test_w1_wa():
     value = random.randint(0, 5)
     original_max = data.array_signal.max()
     data.array_signal **= value
-    assert data.array_signal.max() == original_max**value
+    assert np.isclose(data.array_signal.max(), original_max**value)
     data.close()
 
 

diff --git a/tests/dataset/symmetric_root.py b/tests/dataset/symmetric_root.py
@@ -22,7 +22,7 @@ def test_d1_d2():
     original = data.ai0[:]
     data.ai0.symmetric_root(root)
     assert np.array_equal(np.sign(data.ai0[:]), np.sign(original))
-    assert original.max() ** (1 / root) == data.ai0.max()
+    assert np.isclose(original.max() ** (1 / root), data.ai0.max())
     data.close()
 
 
@@ -33,7 +33,7 @@ def test_w1():
     original = data.signal[:]
     data.signal.symmetric_root(root)
     assert np.array_equal(np.sign(data.signal[:]), np.sign(original))
-    assert original.max() ** (1 / root) == data.signal.max()
+    assert np.isclose(original.max() ** (1 / root), data.signal.max())
     data.close()
 
 
@@ -44,7 +44,7 @@ def test_w1_wa():
     original = data.array_signal[:]
     data.array_signal.symmetric_root(root)
     assert np.array_equal(np.sign(data.array_signal[:]), np.sign(original))
-    assert original.max() ** (1 / root) == data.array_signal.max()
+    assert np.isclose(original.max() ** (1 / root), data.array_signal.max())
     data.close()