Check for an Armstrong number in Python