input: Write a python function to calculate the sum of the squares of two numbers.
  assert sum_squares(2, 4) == 20
  assert sum_squares(3, 5) == 34
  assert sum_squares(0, 7) == 49
output: def sum_squares(a, b):
      return a**2 + b**2