Fix assertions

tskit-dev · Sep 1, 2023 · 32c82a8 · 32c82a8
1 parent 4899426
commit 32c82a8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/tests/beagle.py b/python/tests/beagle.py
@@ -428,10 +428,10 @@ def compute_state_probability_matrix(fm, bm, ref_h, query_h):
     assert m == len(query_h)
     assert (m, h) == fm.shape
     assert (m, h) == bm.shape
-    assert np.any(fm < 0), "Forward probability matrix has negative values."
-    assert np.any(np.isnan(fm)), "Forward probability matrix has NaN values."
-    assert np.any(bm < 0), "Backward probability matrix has negative values."
-    assert np.any(np.isnan(bm)), "Backward probability matrix has NaN values."
+    assert not np.any(fm < 0), "Forward probability matrix has negative values."
+    assert not np.any(np.isnan(fm)), "Forward probability matrix has NaN values."
+    assert not np.any(bm < 0), "Backward probability matrix has negative values."
+    assert not np.any(np.isnan(bm)), "Backward probability matrix has NaN values."
     sm = np.zeros((m, h), dtype=np.float64)
     for i in np.arange(m - 1, -1, -1):
         for j in np.arange(h):