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