input: Write a python function to calculate the sum of squares of two numbers.
  assert sum_of_squares(2, 3) == 13
  assert sum_of_squares(4, 5) == 41
  assert sum_of_squares(0, 0) == 0
output: def sum_of_squares(a, b):
      return a**2 + b**2