Merge pull request #630 from qiboteam/tspatol

Add atol in TSP test
qiboteam · Aug 5, 2022 · 40802bd · 40802bd
2 parents 147d868 + 771bc35
commit 40802bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/qibo/tests/test_models_tsp.py b/src/qibo/tests/test_models_tsp.py
@@ -38,8 +38,8 @@ def qaoa_function_of_layer(backend, layer):
 def test_tsp(backend, nlayers):
     final_state = backend.to_numpy(qaoa_function_of_layer(backend, nlayers))
     assert_regression_fixture(
-        backend, final_state.real, f"tsp_layer{nlayers}_real.out", rtol=1e-3
+        backend, final_state.real, f"tsp_layer{nlayers}_real.out", rtol=1e-3, atol=1e-5
     )
     assert_regression_fixture(
-        backend, final_state.imag, f"tsp_layer{nlayers}_imag.out", rtol=1e-3
+        backend, final_state.imag, f"tsp_layer{nlayers}_imag.out", rtol=1e-3, atol=1e-5
     )
diff --git a/src/qibo/tests/test_models_variational.py b/src/qibo/tests/test_models_variational.py
@@ -12,7 +12,7 @@
 REGRESSION_FOLDER = pathlib.Path(__file__).with_name("regressions")
 
 
-def assert_regression_fixture(backend, array, filename, rtol=1e-5):
+def assert_regression_fixture(backend, array, filename, rtol=1e-5, atol=1e-12):
     """Check array matches data inside filename.
 
     Args:
@@ -34,7 +34,7 @@ def load(filename):
         # case not tested in GitHub workflows because files exist
         np.savetxt(filename, array)
         array_fixture = load(filename)
-    backend.assert_allclose(array, array_fixture, rtol=rtol)
+    backend.assert_allclose(array, array_fixture, rtol=rtol, atol=atol)
 
 
 test_names = "method,options,compile,filename"