diff --git a/testing_project/test_logistic.py b/testing_project/test_logistic.py index 53a32f2..2b6ff30 100644 --- a/testing_project/test_logistic.py +++ b/testing_project/test_logistic.py @@ -1,3 +1,4 @@ +import pytest from numpy.testing import assert_allclose from logistic import f @@ -13,15 +14,17 @@ def test_f_corner_cases(): result = f(x, r) assert_allclose(result, expected) -def test_f_generic_cases(): - cases = [ +@pytest.mark.parametrize( + 'x, r, expected', + [ (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) +) +def test_f_generic_cases(x, r, expected): + result = f(x, r) + assert_allclose(result, expected) # Hands on 1 #Add a new test for these generic cases using the for-loop pattern: