num_digits for negative values

vicky1999 · web-flow · commit 2c04f10b6322 · 2021-01-24T14:50:28.000Z
diff --git a/algorithms/maths/num_digits.py b/algorithms/maths/num_digits.py
@@ -5,6 +5,7 @@
 import math
 
 def num_digits(n):
+  n=abs(n)
   if(n==0):
       return 1;
   return int(math.log10(n))+1
diff --git a/tests/test_maths.py b/tests/test_maths.py
@@ -461,6 +461,9 @@ def test_num_digits(self):
         self.assertEqual(2,num_digits(12))
         self.assertEqual(5,num_digits(99999))
         self.assertEqual(1,num_digits(8))
+        self.assertEqual(1,num_digits(0))        
+        self.assertEqual(1,num_digits(-5))
+        self.assertEqual(3,num_digits(-254))
 
 
 if __name__ == "__main__":
