titaniumtown/YTBN-Graphing-Software
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix newtons_method handling of non-finite derivatives

Browse Source
This commit is contained in:
Simon Gardling 2022-05-18 15:01:33 -04:00
parent d665b72a4c
commit 782d567302
2 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/misc.rs
View File

@ -253,8 +253,14 @@ pub fn newtons_method(
) -> Option<f64> {
let mut x1: f64 = *start_x;
let mut x2: f64;
let mut derivative: f64;
loop {
x2 = x1 - (f(x1) / f_1(x1));
derivative = f_1(x1);
if !derivative.is_finite() {
return None;
}
x2 = x1 - (f(x1) / derivative);
if !x2.is_finite() | !range.contains(&x2) {
return None;
}

18
tests/misc.rs
View File

@ -201,4 +201,22 @@ fn newtons_method() {
&f64::EPSILON,
);
assert_eq!(data, None);
let data = newtons_method(
&|_: f64| f64::INFINITY,
&|x: f64| x.sin(),
&0.0,
&(-10.0..10.0),
&f64::EPSILON,
);
assert_eq!(data, None);
let data = newtons_method(
&|x: f64| x.sin(),
&|_: f64| f64::INFINITY,
&0.0,
&(-10.0..10.0),
&f64::EPSILON,
);
assert_eq!(data, None);
}