Use np.allclose for more robust value comparison

quantumlib · Nov 2, 2023 · 9ddcbaa · 9ddcbaa
1 parent bfb58c2
commit 9ddcbaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cirq-core/cirq/experiments/fidelity_estimation_test.py b/cirq-core/cirq/experiments/fidelity_estimation_test.py
@@ -81,7 +81,7 @@ def test_xeb_fidelity(depolarization, estimator):
         f2 = cirq.xeb_fidelity(
             circuit, bitstrings, qubits, amplitudes=amplitudes, estimator=estimator
         )
-        assert np.abs(f - f2) < 2e-6
+        assert np.allclose(f, f2, rtol=1e-6)
 
         fs.append(f)