diff --git a/testing_project/test_logistic.py b/testing_project/test_logistic.py index 7dc29c8..c2ece9f 100644 --- a/testing_project/test_logistic.py +++ b/testing_project/test_logistic.py @@ -1,17 +1,13 @@ from numpy.testing import assert_allclose - +import pytest from logistic import f -def test_f_corner_cases(): +@pytest.mark.parametrize('x,r,expected', [(0, 1.1, 0),(1, 3.7, 0)]) +def test_f_corner_cases(x, r, expected): # Test cases are (x, r, expected) - cases = [ - (0, 1.1, 0), - (1, 3.7, 0), - ] - for x, r, expected in cases: - result = f(x, r) - assert_allclose(result, 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: