diff --git a/ndarray-linalg/tests/solve.rs b/ndarray-linalg/tests/solve.rs index f26ebccc..d069ec7a 100644 --- a/ndarray-linalg/tests/solve.rs +++ b/ndarray-linalg/tests/solve.rs @@ -19,6 +19,26 @@ fn solve_random_t() { assert_close_l2!(&x, &y, 1e-7); } +#[test] +fn solve_factorized() { + let a: Array2 = random((3, 3)); + let ans: Array1 = random(3); + let b = a.dot(&ans); + let f = a.factorize_into().unwrap(); + let x = f.solve_into(b).unwrap(); + assert_close_l2!(&x, &ans, 1e-7); +} + +#[test] +fn solve_factorized_t() { + let a: Array2 = random((3, 3).f()); + let ans: Array1 = random(3); + let b = a.dot(&ans); + let f = a.factorize_into().unwrap(); + let x = f.solve_into(b).unwrap(); + assert_close_l2!(&x, &ans, 1e-7); +} + #[test] fn rcond() { macro_rules! rcond {