input: Write a function to calculate the square root of an integer.
  assert square_root(9) == 3.0
  assert square_root(16) == 4.0
  assert square_root(25) == 5.0
output: import math
  
  def square_root(n):
      return math.sqrt(n)