input: Write a python function to calculate the factorial of a number using the math library.
  assert factorial(5)==120
  assert factorial(0)==1
  assert factorial(3)==6
output: import math
  
  def factorial(n):
      return math.factorial(n)