some passing tests

molpopgen · Aug 18, 2023 · 4b846c7 · 4b846c7
1 parent 8daf0eb
commit 4b846c7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_genetic_value_to_fitness.py b/tests/test_genetic_value_to_fitness.py
@@ -51,8 +51,8 @@ def testPickle(self):
 
         p = pickle.dumps(self.g, -1)
         up = pickle.loads(p)
-        self.assertEqual(up.VS, self.VS)
-        self.assertEqual(up.optimum, self.opt)
+        self.assertEqual(up.optima[0].VS, self.VS)
+        self.assertEqual(up.optima[0].optimum, self.opt)
 
     def test_mapping(self):
         self.assertEqual(self.g.maps_to_fitness, False)
@@ -67,7 +67,7 @@ def setUp(self):
             Opt(when=0, optimum=0.0, VS=1.0),
             Opt(when=100, optimum=1.0, VS=1.0),
         ]
-        self.g = fwdpy11.GSSmo(self.optima)
+        self.g = fwdpy11.GaussianStabilizingSelection.single_trait(self.optima)
 
     def testPickle(self):
         import pickle