nalgebra/tests/linalg/bidiagonal.rs

106 lines
3.3 KiB
Rust
Raw Normal View History

2023-11-09 07:53:42 +08:00
#[cfg(feature = "proptest-support")]
mod proptest_tests {
macro_rules! gen_tests(
2021-03-01 00:52:14 +08:00
($module: ident, $scalar: expr) => {
mod $module {
#[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-03-01 00:52:14 +08:00
proptest! {
#[test]
fn bidiagonal(m in dmatrix_($scalar)) {
let bidiagonal = m.clone().bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(m, &u * d * &v_t, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn bidiagonal_static_5_3(m in matrix5x3_($scalar)) {
let bidiagonal = m.bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(m, &u * d * &v_t, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn bidiagonal_static_3_5(m in matrix3x5_($scalar)) {
let bidiagonal = m.bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(m, &u * d * &v_t, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn bidiagonal_static_square(m in matrix4_($scalar)) {
let bidiagonal = m.bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(m, &u * d * &v_t, epsilon = 1.0e-7))
}
2021-03-01 00:52:14 +08:00
#[test]
fn bidiagonal_static_square_2x2(m in matrix2_($scalar)) {
let bidiagonal = m.bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
2021-03-01 00:52:14 +08:00
prop_assert!(relative_eq!(m, &u * d * &v_t, epsilon = 1.0e-7))
}
}
}
}
);
2019-03-20 05:53:21 +08:00
2023-11-09 07:53:42 +08:00
gen_tests!(complex, complex_f64());
gen_tests!(f64, PROPTEST_F64);
}
2019-03-20 05:53:21 +08:00
#[test]
fn bidiagonal_identity() {
let m = na::DMatrix::<f64>::identity(10, 10);
2019-03-20 05:53:21 +08:00
let bidiagonal = m.clone().bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
assert_eq!(m, &u * d * &v_t);
let m = na::DMatrix::<f64>::identity(10, 15);
2019-03-20 05:53:21 +08:00
let bidiagonal = m.clone().bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
assert_eq!(m, &u * d * &v_t);
let m = na::DMatrix::<f64>::identity(15, 10);
2019-03-20 05:53:21 +08:00
let bidiagonal = m.clone().bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
assert_eq!(m, &u * d * &v_t);
2020-04-06 00:49:48 +08:00
}
2023-11-09 07:53:42 +08:00
#[test]
fn bidiagonal_regression_issue_1313() {
let s = 6.123234e-16_f32;
let mut m = nalgebra::dmatrix![
10.0, 0.0, 0.0, 0.0, -10.0, 0.0, 0.0, 0.0;
s, 10.0, 0.0, 10.0, s, 0.0, 0.0, 0.0;
20.0, -20.0, 0.0, 20.0, 20.0, 0.0, 0.0, 0.0;
];
m.unscale_mut(m.camax());
let bidiagonal = m.clone().bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
let m2 = &u * d * &v_t;
assert_relative_eq!(m, m2, epsilon = 1e-6);
}
#[test]
fn bidiagonal_regression_issue_1313_minimal() {
let s = 6.123234e-17_f32;
let m = nalgebra::dmatrix![
1.0, 0.0, -1.0;
s, 1.0, s;
];
let bidiagonal = m.clone().bidiagonalize();
let (u, d, v_t) = bidiagonal.unpack();
let m2 = &u * &d * &v_t;
assert_relative_eq!(m, m2, epsilon = 1e-6);
}