Merge pull request keon#740 from vicky1999/master

ankit167 · web-flow · commit e48b9acfdd29 · 2021-01-27T18:32:37.000+05:30
finding number of digits using logs created with unit tests
diff --git a/.gitignore b/.gitignore
@@ -13,5 +13,7 @@ __pycache__/
 /*.egg
 # docs
 build/
-
+pythonenv3.8/
 .vscode/
+# Ignoring the virtual Environment when using GitHub Codespaces
+.venv/
diff --git a/algorithms/maths/__init__.py b/algorithms/maths/__init__.py
@@ -18,6 +18,8 @@
 from .find_order_simple import *
 from .find_primitive_root_simple import *
 from .diffie_hellman_key_exchange import *
+from .num_digits import *
 from .power import *
 from .magic_number import *
 from .krishnamurthy_number import *
+
diff --git a/algorithms/maths/num_digits.py b/algorithms/maths/num_digits.py
@@ -0,0 +1,11 @@
+"""
+   num_digits() method will return the number of digits of a number in O(1) time using math.log10() method.
+"""
+
+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
@@ -22,6 +22,8 @@
     magic_number,
     find_order,
     find_primitive_root,
+    alice_private_key, alice_public_key, bob_private_key, bob_public_key, alice_shared_key, bob_shared_key, diffie_hellman_key_exchange,
+    num_digits,
     alice_private_key, alice_public_key, bob_private_key, bob_public_key, alice_shared_key, bob_shared_key,
     diffie_hellman_key_exchange, krishnamurthy_number
 )
@@ -448,5 +450,21 @@ def test_find_order_simple(self):
         self.assertTrue(diffie_hellman_key_exchange(11, 971))
 
 
+class TestNumberOfDigits(unittest.TestCase):
+    """[summary]
+    Test for the file num_digits.py
+
+    Arguments:
+        unittest {[type]} -- [description]
+    """
+    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__":
     unittest.main()
