Decide to run tests for naive diploid Viterbi depending on the size o…

…f the ref. panel
astheeggeggs · Jun 1, 2024 · fe97477 · fe97477
1 parent 391d3c4
commit fe97477
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_non_tree.py b/tests/test_non_tree.py
@@ -427,7 +427,8 @@ def verify(self, ts, scale_mutation_rate, include_ancestors):
             self.assertAllClose(ll_tmp, path_ll_tmp)
             self.assertAllClose(ll_vs, ll_tmp)
 
-            if not include_ancestors:
+            num_ref_haps = H_vs.shape[1]
+            if num_ref_haps <= 100:
                 V_tmp, P_tmp, ll_tmp = vd.forwards_viterbi_dip_naive(
                     n, m, G_vs, s, e_vs, r
                 )