Virtusa interview question

Write a function to identify Armstrong number