use assert_almost_equal

rigetti · Oct 17, 2024 · b1eef4e · b1eef4e
1 parent 1e84d79
commit b1eef4e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/forest/benchmarking/tests/test_direct_fidelity_estimation.py b/forest/benchmarking/tests/test_direct_fidelity_estimation.py
@@ -46,7 +46,7 @@ def test_exhaustive_process_dfe_run(benchmarker: BenchmarkConnection):
         prog += process
 
         expectation = wfnsim.reset().do_program(prog).expectation(setting.observable)
-        assert expectation == 1.
+        assert_almost_equal(expectation, 1.0)
 
 
 def test_exhaustive_state_dfe_run(benchmarker: BenchmarkConnection):
@@ -62,7 +62,7 @@ def test_exhaustive_state_dfe_run(benchmarker: BenchmarkConnection):
         prog += process
 
         expectation = wfnsim.reset().do_program(prog).expectation(setting.observable)
-        assert expectation == 1.
+        assert_almost_equal(expectation, 1.0)
 
 
 def test_exhaustive_process_dfe_analytical(benchmarker: BenchmarkConnection, qvm):