input: Write a function to calculate the square root of the sum of the squares of two numbers.
  assert square_root(3, 4) == 5.0
  assert square_root(6, 8) == 10.0
  assert square_root(5, 12) == 13.0
output: import math
  
  def square_root(a, b):
      return math.sqrt(a**2 + b**2)