input: Write a function to ensure that the input value to the square_root function is an integer and raises a TypeError if it is not. If the input is valid, it should return the square root of the input value.
  
  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(9) == 3.0
  assert square_root(-4) == 2j
  assert square_root(0) == 0.0
  assert square_root(1) == 1.0
output: def square_root(x):
      if not isinstance(x, int):
          raise TypeError('Input must be an integer')
      return x**0.5