input: Write a python function to return the square of a given number using a lambda function.
  assert square_number(5) == 25
  assert square_number(3) == 9
  assert square_number(0) == 0
output: def square_number(n):
      return (lambda x: x * x)(n)