Check Armstrong Number in Python

#Python program to check if the number provided by the user is an Armstrong number or not
# take input from the user
# num = int(input("Enter a number: "))
# initialize sum
sum = 0
# find the sum of the cube of each digit
temp = num
while temp > 0:

digit = temp % 10
sum += digit ** 3
temp //= 10

# display the result
if num == sum

print(num,"is an Armstrong number")

else:

print(num,"is not an Armstrong number")

Output 1:

Enter a number:663
663 is not an Armstrong number

Output 2:

Enter a number:407
407 is an Armstrong number