Write code for an Armstrong number.
Implement is_armstrong(number) to return True when number equals the sum of each digit raised to the total number of digits, and False otherwise. Treat 0 as an Armstrong number. The input is a non-negative integer, and the output is a Boolean.
def is_armstrong(number):