Add rtol to kpz checks (#335)

Add rtol to kpz checks
quantumlib · Sep 1, 2023 · d021621 · d021621
1 parent 7c831ab
commit d021621
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/recirq/kpz/experiment_test.py b/recirq/kpz/experiment_test.py
@@ -126,10 +126,10 @@ def test_size_independence_muinf():
     expt2 = kpz.KPZExperiment(cycles, mu, 1, theta, phi, num_qubits=2 * cycles + 2)
     res2 = expt2.run_experiment_amplitudes(_SAMPLER)
 
-    assert np.isclose(res1.mean, res2.mean, atol=1e-5)
-    assert np.isclose(res1.variance, res2.variance, atol=1e-5)
-    assert np.isclose(res1.skewness, res2.skewness, atol=2e-5)
-    assert np.isclose(res1.kurtosis, res2.kurtosis, atol=2e-5)
+    assert np.isclose(res1.mean, res2.mean, atol=1e-5, rtol=2e-5)
+    assert np.isclose(res1.variance, res2.variance, atol=1e-5, rtol=2e-5)
+    assert np.isclose(res1.skewness, res2.skewness, atol=2e-5, rtol=2e-5)
+    assert np.isclose(res1.kurtosis, res2.kurtosis, atol=2e-5, rtol=2e-5)
 
 
 def test_size_independence():