diff --git a/tests/test_math_utils.py b/tests/test_math_utils.py index e3a29da0e..3043ee279 100644 --- a/tests/test_math_utils.py +++ b/tests/test_math_utils.py @@ -19,9 +19,16 @@ def test_call_missing_function(): # Accessing a missing function attribute raises AttributeError getattr(math_utils, "non_existent_function")() -def test_addition_pass(): - # Test passes: correct addition - assert addition(2, 2) == 4 +@pytest.mark.parametrize( + "a, b, expected", + [ + (2, 2, 4), + (1, 5, 6), + (3, 4, 7), + ] +) +def test_addition_pass(a, b, expected): + assert addition(a, b) == expected def test_addition_fail(): # Assertion failure: expected incorrect result