armstrong number bug fix

avinashn · avinashn · commit e4ef548fd3ce · 2019-03-31T17:33:39.000+05:30
diff --git a/armstrong.py b/armstrong.py
@@ -1,13 +1,14 @@
 __author__ = 'Avinash'
 
 input_num = (input("Enter any number: "))
+digit_len = len(str(input_num))
 
 try:
     arm_num = 0
     val = int(input_num)
     while val > 0:
         reminder = val % 10
-        arm_num += reminder ** 3
+        arm_num += reminder ** digit_len
         val //= 10
 
     if int(input_num) == arm_num:
