test_mathlib.py

importmathlib
importpytest

@pytest.mark.parametrize("test_input, expected_output",
 [
 (5, 25),
 (9, 81),
 (10, 100)
 ]
 )
deftest_calc_square(test_input, expected_output):
result=mathlib.calc_square(test_input)
assertresult==expected_output