"""
    square of number

    function test_square_number
    input --> number

"""

def test_square_number(number):
    return number ** 2


assert test_square_number(2) == 4
print(test_square_number(20))