Use everywhere for saftey

scikit-hep · Jan 13, 2021 · 397a5d7 · 397a5d7
1 parent ebafa43
commit 397a5d7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_probability.py b/tests/test_probability.py
@@ -56,8 +56,10 @@ def test_joint(backend):
     tb, _ = 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])) == tb.tolist(
-        p1 + p2
+    assert np.allclose(
+        tb.tolist(probability.Simultaneous._joint_logpdf([p1, p2])),
+        tb.tolist(p1 + p2),
+        atol=1e-12,
     )