Commit 2822416

authored

Update weierstrass_method.py

* add more tests

1 parent 3c720e5 commit 2822416Copy full SHA for 2822416

File tree

-4

lines changed

-4

lines changed

Lines changed: 9 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -32,9 +32,14 @@ def weierstrass_method(`
`32`	`32`
`33`	`33`	`Example:`
`34`	`34`	`>>> import numpy as np`
`35`		`- >>> def f(x): return x**2 - 1`
`36`		`- >>> roots = weierstrass_method(f, 2)`
`37`		`- >>> np.allclose(np.sort(roots), np.sort(np.array([1, -1])))`
	`35`	`+ >>> def check(poly, degree, expected):`
	`36`	`+ ... roots = weierstrass_method(poly, degree)`
	`37`	`+ ... return np.allclose(np.sort(roots), np.sort(expected))`
	`38`	`+`
	`39`	`+ >>> check(lambda x: x**2 - 1, 2, np.array([-1, 1]))`
	`40`	`+ True`
	`41`	`+`
	`42`	`+ >>> check(lambda x: x*3 - 4.5x*2 + 5.75x - 1.875, 3, np.array([1.5, 0.5, 2.5]))`
`38`	`43`	`True`
`39`	`44`
`40`	`45`	`See Also:`
`@@ -82,4 +87,4 @@ def weierstrass_method(`
`82`	`87`	`if __name__ == "__main__":`
`83`	`88`	`import doctest`
`84`	`89`
`85`		`- doctest.testmod()`
	`90`	`+ doctest.testmod()`

Comments

(0)