input: Write a function that checks if a given number is an Armstrong number. An Armstrong number (also known as narcissistic numbers) is a number that is the sum of its own digits each raised to the power of the number of digits. For example, 153 is an Armstrong number because 1^3 + 5^3 + 3^3 = 153.
  
  def is_armstrong_number(num):
      n = len(str(num))
      temp = num
      sum_of_digits = 0
  
      while temp > 0:
          digit = temp % 10
          sum_of_digits += digit ** n
          temp //= 10
  
      if num == sum_of_digits:
          return True
      else:
          return False
  
  assert is_armstrong_number(153) == True
  assert is_armstrong_number(9474) == True
  assert is_armstrong_number(123) == False
  assert is_armstrong_number(0) == True
  assert is_armstrong_number(1) == True
output: def is_armstrong_number(num):
      n = len(str(num))
      temp = num
      sum_of_digits = 0
  
      while temp > 0:
          digit = temp % 10
          sum_of_digits += digit ** n
          temp //= 10
  
      if num == sum_of_digits:
          return True
      else:
          return False