Change tolerance test

3 years ago · 53a6bfebe9
parent cceba9a83c
commit 53a6bfebe9
1 changed files with 1 additions and 2 deletions
--- a/tests/stages/generation/test.py
+++ b/tests/stages/generation/test.py
@ -37,8 +37,7 @@ def generate(N):
 def test(N):
    k = generate(N)
    k_correct = np.load(f"out_{N}.npy")
-    comparison = k == k_correct
+    return np.allclose(k, k_correct, rtol=1e-10, atol=1e-10)
    return comparison.all()
 class TestGeneration(unittest.TestCase):
    def test_8(self):