Use BinaryMutationModel to ensure biallelic sites

astheeggeggs · Jun 21, 2024 · 75b8525 · 75b8525
1 parent c7f2125
commit 75b8525
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/lsbase.py b/tests/lsbase.py
@@ -204,6 +204,7 @@ def get_ts_simple_n10_no_recomb(self, seed=42):
                 recombination_rate=0.0,
                 random_seed=seed,
             ),
+            model=msprime.BinaryMutationModel(),
             rate=0.5,
             random_seed=seed,
         )
@@ -220,6 +221,7 @@ def get_ts_simple(self, num_samples, seed=42):
                 random_seed=seed,
             ),
             rate=5.0,
+            model=msprime.BinaryMutationModel(),
             random_seed=seed,
         )
         assert ts.num_sites > 5
@@ -235,6 +237,7 @@ def get_ts_simple_n8_high_recomb(self, seed=42):
                 random_seed=seed,
             ),
             rate=5.0,
+            model=msprime.BinaryMutationModel(),
             random_seed=seed,
         )
         assert ts.num_trees > 15
@@ -251,6 +254,7 @@ def get_ts_custom_pars(self, num_samples, seq_length, mean_r, mean_mu, seed=42):
                 random_seed=seed,
             ),
             rate=mean_mu,
+            model=msprime.BinaryMutationModel(),
             random_seed=seed,
         )
         return ts