96ffd64c0c
- used `syn::parse_quote!` instead of constructing `syn` structs directly. unsurprisingly, makes a world of difference! - code is, like, 1k lines shorter now - moved tests into dedicated crate - tweaked the tests even more, to represent complete lack of dependency on external crates
166 lines
4.2 KiB
Rust
166 lines
4.2 KiB
Rust
#![allow(missing_docs)]
|
|
|
|
#[test]
|
|
fn array() {
|
|
use approx::assert_ulps_eq;
|
|
use nacfahi::{models::basic::Linear, *};
|
|
|
|
// some data, presumably 2x + 1
|
|
let x = [1.0, 2.0, 3.0, 4.0, 5.0];
|
|
let y = [3.0, 5.0, 7.0, 9.0, 11.0];
|
|
|
|
// fitting model: a*x + b
|
|
let mut line = Linear { a: 0.0, b: 0.0 };
|
|
|
|
// do the fit!
|
|
let report = fit!(&mut line, x, y);
|
|
|
|
// check that approximation is successful
|
|
assert!(
|
|
report.termination.was_successful(),
|
|
"Approximation should be successful"
|
|
);
|
|
|
|
// check that model parameters have expected values
|
|
assert_ulps_eq!(line.a, 2.0);
|
|
assert_ulps_eq!(line.b, 1.0);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(feature = "alloc")]
|
|
fn slice() {
|
|
use approx::assert_ulps_eq;
|
|
use nacfahi::{models::basic::Linear, *};
|
|
|
|
// some data, presumably 2x + 1
|
|
let x = &[1.0, 2.0, 3.0, 4.0, 5.0][..];
|
|
let y = &[3.0, 5.0, 7.0, 9.0, 11.0][..];
|
|
|
|
// fitting model: a*x + b
|
|
let mut line = Linear { a: 0.0, b: 0.0 };
|
|
|
|
// do the fit!
|
|
let report = fit!(&mut line, x, y);
|
|
|
|
// check that approximation is successful
|
|
assert!(
|
|
report.termination.was_successful(),
|
|
"Approximation should be successful"
|
|
);
|
|
|
|
// check that model parameters have expected values
|
|
assert_ulps_eq!(line.a, 2.0);
|
|
assert_ulps_eq!(line.b, 1.0);
|
|
}
|
|
|
|
#[cfg(feature = "nalgebra")]
|
|
#[test]
|
|
fn matrix() {
|
|
use approx::assert_ulps_eq;
|
|
use nacfahi::{models::basic::Linear, *};
|
|
use nalgebra::matrix;
|
|
|
|
// some data, presumably 2x + 1
|
|
let x = matrix![1.0; 2.0; 3.0; 4.0; 5.0];
|
|
let y = matrix![3.0; 5.0; 7.0; 9.0; 11.0];
|
|
|
|
// fitting model: a*x + b
|
|
let mut line = Linear { a: 0.0, b: 0.0 };
|
|
|
|
// do the fit!
|
|
let report = fit!(&mut line, x, y);
|
|
|
|
// check that approximation is successful
|
|
assert!(
|
|
report.termination.was_successful(),
|
|
"Approximation should be successful"
|
|
);
|
|
|
|
// check that model parameters have expected values
|
|
assert_ulps_eq!(line.a, 2.0);
|
|
assert_ulps_eq!(line.b, 1.0);
|
|
}
|
|
|
|
#[cfg(feature = "nalgebra")]
|
|
#[test]
|
|
fn matrix_view() {
|
|
use approx::assert_ulps_eq;
|
|
use nacfahi::{models::basic::Linear, *};
|
|
use nalgebra::matrix;
|
|
|
|
// some data, presumably 2x + 1
|
|
let x = matrix![1.0; 2.0; 3.0; 4.0; 5.0];
|
|
let y = matrix![3.0; 5.0; 7.0; 9.0; 11.0];
|
|
|
|
// fitting model: a*x + b
|
|
let mut line = Linear { a: 0.0, b: 0.0 };
|
|
|
|
// do the fit!
|
|
let report = fit!(&mut line, x.column(0), y.column(0));
|
|
|
|
// check that approximation is successful
|
|
assert!(
|
|
report.termination.was_successful(),
|
|
"Approximation should be successful"
|
|
);
|
|
|
|
// check that model parameters have expected values
|
|
assert_ulps_eq!(line.a, 2.0);
|
|
assert_ulps_eq!(line.b, 1.0);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg(all(feature = "nalgebra", feature = "alloc"))]
|
|
fn dyn_matrix() {
|
|
use approx::assert_ulps_eq;
|
|
use nacfahi::{models::basic::Linear, *};
|
|
use nalgebra::{Dyn, Vector};
|
|
|
|
// some data, presumably 2x + 1
|
|
let x = Vector::<_, Dyn, _>::from_vec(vec![1.0, 2.0, 3.0, 4.0, 5.0]);
|
|
let y = x.map(|x| 2.0 * x + 1.0);
|
|
|
|
// fitting model: a*x + b
|
|
let mut line = Linear { a: 0.0, b: 0.0 };
|
|
|
|
// do the fit!
|
|
let report = fit!(&mut line, x, y);
|
|
|
|
// check that approximation is successful
|
|
assert!(
|
|
report.termination.was_successful(),
|
|
"Approximation should be successful"
|
|
);
|
|
|
|
// check that model parameters have expected values
|
|
assert_ulps_eq!(line.a, 2.0);
|
|
assert_ulps_eq!(line.b, 1.0);
|
|
}
|
|
|
|
#[cfg(feature = "generic-array")]
|
|
#[test]
|
|
fn generic_array() {
|
|
use approx::assert_ulps_eq;
|
|
use generic_array::GenericArray;
|
|
use nacfahi::{models::basic::Linear, *};
|
|
|
|
// some data, presumably 2x + 1
|
|
let x = GenericArray::from_array([1.0, 2.0, 3.0, 4.0, 5.0]);
|
|
let y = GenericArray::from_array([3.0, 5.0, 7.0, 9.0, 11.0]);
|
|
|
|
// fitting model: a*x + b
|
|
let mut line = Linear { a: 0.0, b: 0.0 };
|
|
|
|
// do the fit!
|
|
let report = fit!(&mut line, x, y);
|
|
|
|
// check that approximation is successful
|
|
assert!(
|
|
report.termination.was_successful(),
|
|
"Approximation should be successful"
|
|
);
|
|
|
|
// check that model parameters have expected values
|
|
assert_ulps_eq!(line.a, 2.0);
|
|
assert_ulps_eq!(line.b, 1.0);
|
|
}
|