input: Write a python function to calculate the area of a circle given its radius.
  assert area(3) == 28.274333882308138
  assert area(5) == 78.53981633974483
  assert area(0) == 0
output: import math
  
  def area(radius):
      return math.pi * radius ** 2