input: Write a function to ensure that the input value in the `square_root` function is an integer. If the input value is not an integer, raise a `TypeError`. If the input value is an integer, return its square root.
  
  def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5
  
  assert square_root(4)==2.0
  assert square_root(-1)==raises TypeError
  assert square_root('a')==raises TypeError
  assert square_root(9)==3.0
output: def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5