type error check

Author: mahiuddin-dev

maths/number_of_digits.py

@@ -16,7 +16,15 @@ def num_digits(n: int) -> int:
     1
     >>> num_digits(-123456)
     6
+    >>> num_digits('123')  # Raises a TypeError for non-integer input
+    Traceback (most recent call last):
+        ...
+    TypeError: Input must be an integer
     """
+    
+    if not isinstance(n, int):
+        raise TypeError("Input must be an integer")
+
     digits = 0
     n = abs(n)
     while True:
@@ -42,7 +50,15 @@ def num_digits_fast(n: int) -> int:
     1
     >>> num_digits_fast(-123456)
     6
+    >>> num_digits('123')  # Raises a TypeError for non-integer input
+    Traceback (most recent call last):
+        ...
+    TypeError: Input must be an integer
     """
+    
+    if not isinstance(n, int):
+        raise TypeError("Input must be an integer")
+    
     return 1 if n == 0 else math.floor(math.log(abs(n), 10) + 1)
 
 
@@ -61,7 +77,15 @@ def num_digits_faster(n: int) -> int:
     1
     >>> num_digits_faster(-123456)
     6
+    >>> num_digits('123')  # Raises a TypeError for non-integer input
+    Traceback (most recent call last):
+        ...
+    TypeError: Input must be an integer
     """
+    
+    if not isinstance(n, int):
+        raise TypeError("Input must be an integer")
+    
     return len(str(abs(n)))
 
 
@@ -76,7 +100,7 @@ def benchmark_a_function(func: Callable, value: int) -> None:
         timing = timeit(f"__main__.{call}", setup="import __main__")
         print(f"{call}: {func(value)} -- {timing} seconds")
 
-    for value in (262144, 1125899906842624, 1267650600228229401496703205376):
+    for value in ('262144', 1125899906842624, 1267650600228229401496703205376):
         for func in (num_digits, num_digits_fast, num_digits_faster):
             benchmark_a_function(func, value)
         print()