For a given 3 digit number, find whether it is armstrong number or not

Armstrong Numbers

# https://practice.geeksforgeeks.org/problems/armstrong-numbers/0/?track=interview-mathematical

import os
import random
import re
import sys

t = int(input())
for i in range(t):
    n = int(input())
    temp = n
    sum = 0
    while(temp>0):
        digit = int(temp%10)
        #print("digit is: ",digit)
        sum+=(digit*digit*digit)
        #print("sum is: ",sum)
        temp = int(temp/10)
        #print("temp  is: ",temp)
    #print(sum)
    if sum == n:
        print("Yes")
    else:
        print("No")