Checking if a number is prime

Write a function is_prime which receives an integer as a parameter and returns a boolean. The function will return True when the number is a prime, otherwise it will return False).

You can use the following pytest to test your function.

@pytest.mark.parametrize("testcase, input, expected_output",[(1, 0, False), (2, 1, False), (3, 2, True), (4, 25, False), (5, 23, True), (6, 97, True) ] )

def test_is_prime(testcase, input, expected_output): 
    assert is_prime(input) == expected_output, "case 0".format(testcase)

Insist that the students test their programs by giving them example
pytests.