input: Write a python function to calculate the square root of an integer and return the result.
  assert square_root(9) == 3.0
  assert square_root(16) == 4.0
  assert square_root(25) == 5.0
output: def square_root(n):
      return n ** 0.5