input: Write a python function to calculate the square of a given number.
  assert square_of_number(2) == 4
  assert square_of_number(5) == 25
  assert square_of_number(10) == 100
output: def square_of_number(n):
      return n * n