Use allclose with tolerance of 1e-12

scikit-hep · Jan 13, 2021 · ebafa43 · ebafa43
1 parent 8735ce5
commit ebafa43
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_probability.py b/tests/test_probability.py
@@ -69,8 +69,10 @@ def test_independent(backend):
 
     p1 = probability.Poisson(tb.astensor([10.0])).log_prob(tb.astensor(2.0))
     p2 = probability.Poisson(tb.astensor([10.0])).log_prob(tb.astensor(3.0))
-    assert tb.tolist(probability.Simultaneous._joint_logpdf([p1, p2]))[0] == tb.tolist(
-        result
+    assert np.allclose(
+        tb.tolist(probability.Simultaneous._joint_logpdf([p1, p2]))[0],
+        tb.tolist(result),
+        atol=1e-12,
     )