input: Write a function to modify the given Python code so that it raises an exception if the input value is not an integer. The function should be named `square_root`. If the input is not an integer, a `TypeError` should be raised with the message 'Input must be an integer'. If the input is an integer, the function should 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(16)==4.0
  try:
      square_root('4')
      assert False, "Expected TypeError"
  except TypeError:
      pass
  try:
      square_root(4.0)
      assert False, "Expected TypeError"
  except TypeError:
      pass
output: def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5