@ -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):