Python Program to Check Armstrong Number

n=int(input("Enter a number :")
s=0
q=n
while(n>0):
rem=n%10
s=s+(rem*rem*rem)
n=n/10
if(s==q):
print("It is Armstrong number!")
else:
print("It is not Armstrong number!")