testing_project/test_logistic_fit.py

@@ -1,4 +1,5 @@
 from numpy.testing import assert_allclose
+import numpy as np
 
 from logistic import iterate_f
 from logistic_fit import fit_r
@@ -9,3 +10,13 @@ def test_fit_r():
     fit_result_r = fit_r(trajectory)
     assert_allclose(r, fit_result_r)
 
+
+def test_fit_r_randomized():
+    random_state = np.random.RandomState(42)
+    x0 = 0.3
+    for _ in range(100):
+        r = random_state.rand()
+        trajectory = iterate_f(x0, r, 23)
+        fit_result_r = fit_r(trajectory)
+        assert_allclose(r, fit_result_r, atol=0.1)
+