nalgebra/tests/linalg/solve.rs

67 lines
2.4 KiB
Rust
Raw Normal View History

2021-03-01 00:52:14 +08:00
#![cfg(feature = "proptest-support")]
macro_rules! gen_tests(
2021-03-01 00:52:14 +08:00
($module: ident, $scalar: expr) => {
mod $module {
2021-03-01 00:52:14 +08:00
use na::{Matrix4, ComplexField};
#[allow(unused_imports)]
2019-03-23 21:29:07 +08:00
use crate::core::helper::{RandScalar, RandComplex};
2021-03-01 00:52:14 +08:00
use crate::proptest::*;
use proptest::{prop_assert, proptest};
2021-04-11 17:00:38 +08:00
fn unzero_diagonal<T: ComplexField>(a: &mut Matrix4<T>) {
for i in 0..4 {
2021-08-04 23:56:57 +08:00
if a[(i, i)].clone().norm1() < na::convert(1.0e-7) {
2021-04-11 17:00:38 +08:00
a[(i, i)] = T::one();
}
}
}
2021-03-01 00:52:14 +08:00
proptest! {
#[test]
fn solve_lower_triangular(a in matrix4_($scalar), b in matrix4x5_($scalar)) {
let mut a = a;
unzero_diagonal(&mut a);
let tri = a.lower_triangle();
let x = a.solve_lower_triangular(&b).unwrap();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(tri * x, b, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn solve_upper_triangular(a in matrix4_($scalar), b in matrix4x5_($scalar)) {
let mut a = a;
unzero_diagonal(&mut a);
let tri = a.upper_triangle();
let x = a.solve_upper_triangular(&b).unwrap();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(tri * x, b, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn tr_solve_lower_triangular(a in matrix4_($scalar), b in matrix4x5_($scalar)) {
let mut a = a;
unzero_diagonal(&mut a);
let tri = a.lower_triangle();
let x = a.tr_solve_lower_triangular(&b).unwrap();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(tri.transpose() * x, b, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn tr_solve_upper_triangular(a in matrix4_($scalar), b in matrix4x5_($scalar)) {
let mut a = a;
unzero_diagonal(&mut a);
let tri = a.upper_triangle();
let x = a.tr_solve_upper_triangular(&b).unwrap();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(tri.transpose() * x, b, epsilon = 1.0e-7))
}
}
}
}
);
2021-03-01 00:52:14 +08:00
gen_tests!(complex, complex_f64());
gen_tests!(f64, PROPTEST_F64);