input: Write a function to calculate the power of a number.
  assert power(2, 3) == 8
  assert power(5, 0) == 1
  assert power(7, 2) == 49
  assert power(10, 1) == 10
output: def power(base, exponent):
      return base ** exponent