### Not mine solution
def square_num(numb):
    return numb**2

def test_square_num():
	assert square_num(2) == 4
	assert square_num(8) == 64
	assert square_num(10) == 100
	print('Your code is correct')

test_square_num()