add a test for when seed=0

cjekel · Dec 31, 2024 · f192f2b · f192f2b
1 parent 1bb400a
commit f192f2b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/tests.py b/tests/tests.py
@@ -544,13 +544,14 @@ def test_random_seed_fit(self):
         self.assertTrue(same_breaks.sum() == same_breaks.size)
 
     def test_random_seed_fitfast(self):
+        # specifically test for seed = 0
         np.random.seed(1)
         my_pwlf = pwlf.PiecewiseLinFit(self.x_small, self.y_small,
-                                       seed=123)
+                                       seed=0)
         fit1 = my_pwlf.fitfast(2)
         np.random.seed(2)
         my_pwlf = pwlf.PiecewiseLinFit(self.x_small, self.y_small,
-                                       seed=123)
+                                       seed=0)
         fit2 = my_pwlf.fitfast(2)
         same_breaks = np.isclose(fit1, fit2)
         self.assertTrue(same_breaks.sum() == same_breaks.size)