Fix bug

Author: szhan

python/tests/test_beagle.py

@@ -274,8 +274,8 @@ def compute_state_probability_matrix(fm, bm, ref_h, query_h, rho, mu):
     assert fm.shape == (m, h)
     assert bm.shape == (m, h)
     # Check all biallelic sites
-    assert np.unique(ref_h) == np.array([0, 1])
-    assert np.unique(query_h) == np.array([0, 1])
+    assert np.all(np.isin(np.unique(ref_h), [0, 1]))
+    assert np.all(np.isin(np.unique(query_h), [-1, 0, 1]))
     sm = np.zeros((m, h), dtype=np.float64)  # HMM state probability matrix
     fwd_hap_probs = np.zeros((m, 2), dtype=np.float64)
     bwd_hap_probs = np.zeros((m, 2), dtype=np.float64)