Число Армстронга

Вихідний код

def is_armstrong(n):
    s = 0
    t = n
    while t > 0:
        d = t % 10
        s += d * d * d
        t //= 10
    if s == n:
        print("Так")
    else:
        print("Ні")