added type hints and doctests to arithmetic_analysis/newton_method.py

Added type hints and doctests to arithmetic_analysis/newton_method.py
TheAlgorithms · cclauss · Aug 1, 2020 · cclauss · Aug 1, 2020 · spamegg1
diff --git a/arithmetic_analysis/newton_method.py b/arithmetic_analysis/newton_method.py
+
+# function is the f(x) and derivative is the f'(x)
+def newton(
+    function: Callable[[float], float],
diff --git a/arithmetic_analysis/newton_method.py b/arithmetic_analysis/newton_method.py
+        if derivative(prev_guess) == 0:
+            raise ZeroDivisionError("float division by zero, could not find root")
+
+        next_guess: float = prev_guess - function(prev_guess) / derivative(prev_guess)
-        if derivative(prev_guess) == 0:
-            raise ZeroDivisionError("float division by zero, could not find root")
-
-        next_guess: float = prev_guess - function(prev_guess) / derivative(prev_guess)
+        try:
+            next_guess: float = prev_guess - function(prev_guess) / derivative(prev_guess)
+        except ZeroDivisionError:
+            raise ZeroDivisionError("Could not find root")
diff --git a/arithmetic_analysis/newton_method.py b/arithmetic_analysis/newton_method.py
+            if abs(prev_guess - next_guess) < 10 ** -5:
+                return next_guess
+            prev_guess = next_guess