def test_square_by_parametrize(number):
assert square(number) == number**2
def test_square_with_positive_input():
assert square(3) == 9
def test_square_with_a_string():
with pytest.raises(TypeError):
assert square("s")
def test_square_with_zero():
assert square(0) == 0
def test_square_with_negative_input():
assert square(-4) == 16
def test_square():
assert(sample.square(2) == 4)