From 9c86af7cee62d6a72bb2437ea40c09f7b41e8803 Mon Sep 17 00:00:00 2001 From: Toshiki Teramura Date: Wed, 29 Jul 2020 02:35:06 +0900 Subject: [PATCH] To confirm #213 resolved #228 --- ndarray-linalg/tests/solve.rs | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) 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 {