diff --git a/logistic.py b/logistic.py index e49d1c2..5382c65 100644 --- a/logistic.py +++ b/logistic.py @@ -1 +1,3 @@ -# Your code goes here +def f(x , r): + """Compute the logistic map for a given value of x and r""" + return r * x * (1 - x) diff --git a/test_logistic.py b/test_logistic.py index 9391bee..142f64e 100644 --- a/test_logistic.py +++ b/test_logistic.py @@ -4,7 +4,6 @@ # Add here your test for the logistic map - def test_f_corner_cases(): # Test cases are (x, r, expected) cases = [ @@ -14,3 +13,15 @@ def test_f_corner_cases(): for x, r, expected in cases: result = f(x, r) assert_allclose(result, expected) + + +def test_f_generic_cases(): + # Test cases are (x, r, expected) + cases = [ + (0.1, 2.2, 0.198), + (0.2, 3.4, 0.544), + (0.5, 2, 0.5) + ] + for x, r, expected in cases: + result = f(x, r) + assert_allclose(result, expected)