input: Write a function to calculate the average of five numbers entered by a user. If the sum of the numbers is zero, return 0.
  
  def calculate_average():
      nums = []
      for i in range(5):
          nums.append(int(input("Please enter number: ")))
      
      # Calculate average
      average = sum(nums) / len(nums) if sum(nums) != 0 else 0
      
      # Print average
      print("The average is ", average)
  
  assert calculate_average() == (sum([1, 2, 3, 4, 5]) / 5)
  assert calculate_average() == 0
output: def calculate_average():
      nums = []
      for i in range(5):
          nums.append(int(input("Please enter number: ")))
      return sum(nums) / len(nums) if sum(nums) != 0 else 0