Help me in solving LPYAS151 problem

help
,
1

My issue

the test cases are not passing pls check for that

My code

n = int(input())
sum=0
temp=n
while temp>0:
    digit = temp%10
    sum+=digit**3
    temp//=10
if n==sum:
    print('Armstrong')
else:
    print('Not Armstrong')

Learning course: Roadmap to 3*
Problem Link: https://www.codechef.com/learn/course/klu-roadmap-3star/KLURMP300B/problems/LPYAS151