From 18a2d52f2b0ec4a1d822f2f89eca3f7538b3be7a Mon Sep 17 00:00:00 2001
From: chortas <chortas@fi.uba.ar>
Date: Mon, 24 Jan 2022 18:20:48 -0300
Subject: [PATCH] Change tolerance test

---
 tests/stages/generation/test.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/stages/generation/test.py b/tests/stages/generation/test.py
index 732fdf1..357f542 100644
--- a/tests/stages/generation/test.py
+++ b/tests/stages/generation/test.py
@@ -37,9 +37,8 @@ def generate(N):
 def test(N):
     k, duration = generate(N)
     k_correct = np.load(f"out_{N}.npy")
-    comparison = k == k_correct
     print(f"Generation with N = {N} time = {duration}s")
-    return comparison.all()
+    return np.allclose(k, k_correct, rtol=1e-10, atol=1e-10)
 
 class TestGeneration(unittest.TestCase):
     def test_08(self):