Pass parameters

szhan · Sep 13, 2023 · eb7e20a · eb7e20a
1 parent 49b9f51
commit eb7e20a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/tests/test_imputation.py b/python/tests/test_imputation.py
@@ -546,8 +546,12 @@ def test_tsimpute(input_ref, input_query):
         imputed_alleles, allele_probs = tests.beagle.run_beagle(
             input_ref, input_query[i], pos, miscall_rate=0.0001, ne=10.0
         )
+        # Note that parametrization of BEAGLE and tsinfer
+        mu = tests.beagle_numba.get_mismatch_prob(pos, miscall_rate=1e-8)
+        rho = tests.beagle_numba.get_switch_prob(pos, input_ref.shape[0], ne=10_000.0)
+        rho /= 1e5
         imputed_alleles_ts, allele_probs_ts = tests.beagle_numba.run_tsimpute(
-            input_ref, input_query[i], pos
+            input_ref, input_query[i], pos, mu, rho
         )
         assert np.array_equal(imputed_alleles, imputed_alleles_ts)
         assert np.allclose(allele_probs, allele_probs_ts)