From 53a6bfebe914e47b2f006e0ff1cbc7b2435b9470 Mon Sep 17 00:00:00 2001
From: chortas <chortas@fi.uba.ar>
Date: Mon, 24 Jan 2022 18:19:17 -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 94118b1..5158d78 100644
--- 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 comparison.all()
+    return np.allclose(k, k_correct, rtol=1e-10, atol=1e-10)
 
 class TestGeneration(unittest.TestCase):
     def test_8(self):