Update weierstrass_method.py

Author: aditya7balotra

maths/numerical_analysis/weierstrass_method.py

@@ -36,10 +36,17 @@ def weierstrass_method(
         ...     roots = weierstrass_method(poly, degree)
         ...     return np.allclose(np.sort(roots), np.sort(expected))
 
-        >>> check(lambda x: x**2 - 1, 2, np.array([-1, 1]))
+        >>> check(
+        ...     lambda x: x**2 - 1,
+        ...     2,
+        ...     np.array([-1, 1]))
         True
 
-        >>> check(lambda x: x**3 - 4.5*x**2 + 5.75*x - 1.875, 3, np.array([1.5, 0.5, 2.5]))
+        >>> check(
+        ...     lambda x: x**3 - 4.5*x**2 + 5.75*x - 1.875,
+        ...     3,
+        ...     np.array([1.5, 0.5, 2.5])
+        ... )
         True
 
     See Also: