fix test setup

molpopgen · Nov 13, 2023 · 76b065f · 76b065f
1 parent 85dc23f
commit 76b065f
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/tests_with_cpp/test_custom_stateless_fitness.py b/tests_with_cpp/test_custom_stateless_fitness.py
@@ -13,14 +13,13 @@ class testCustomAdditive(unittest.TestCase):
     def setUp(self):
         self.pop = fwdpy11.DiploidPopulation(1000, 1.0)
         self.pdict = mslike(
-            self.pop, dfe=fwdpy11.ExpS(0, 1, 1, -0.05),
-            pneutral=0.95, simlen=10
+            self.pop, dfe=fwdpy11.ExpS(0, 1, 1, -0.05), pneutral=0.95, simlen=10
         )
+        self.pdict["prune_selected"] = False
         self.pdict["gvalue"] = ca.additive()
-        self.pdict["demography"] = \
-            fwdpy11.ForwardDemesGraph.tubes(self.pop.deme_sizes()[1],
-                                            burnin=self.pdict["simlen"],
-                                            burnin_is_exact=True)
+        self.pdict["demography"] = fwdpy11.ForwardDemesGraph.tubes(
+            self.pop.deme_sizes()[1], burnin=self.pdict["simlen"], burnin_is_exact=True
+        )
         self.rng = fwdpy11.GSLrng(42)
         self.params = fwdpy11.ModelParams(**self.pdict)
 
@@ -44,11 +43,11 @@ def setUp(self):
             pneutral=0.95,
             simlen=10,
         )
+        self.pdict["prune_selected"] = False
         self.pdict["gvalue"] = general.GeneralW()
-        self.pdict["demography"] =\
-            fwdpy11.ForwardDemesGraph.tubes(self.pop.deme_sizes()[1],
-                                            burnin=self.pdict["simlen"],
-                                            burnin_is_exact=True)
+        self.pdict["demography"] = fwdpy11.ForwardDemesGraph.tubes(
+            self.pop.deme_sizes()[1], burnin=self.pdict["simlen"], burnin_is_exact=True
+        )
         self.rng = fwdpy11.GSLrng(42)
         self.params = fwdpy11.ModelParams(**self.pdict)