Add test for solve Ax=b

Author: termoshtt

tests/solve.rs

@@ -139,3 +139,21 @@ fn det_nonsquare() {
         }
     }
 }
+
+#[test]
+fn solve_random() {
+    let a: Array2<f64> = random((3, 3));
+    let x: Array1<f64> = random(3);
+    let b = a.dot(&x);
+    let y = a.solve_into(b).unwrap();
+    assert_close_l2!(&x, &y, 1e-7);
+}
+
+#[test]
+fn solve_random_t() {
+    let a: Array2<f64> = random((3, 3).f());
+    let x: Array1<f64> = random(3);
+    let b = a.dot(&x);
+    let y = a.solve_into(b).unwrap();
+    assert_close_l2!(&x, &y, 1e-7);
+}

tests/solveh.rs

@@ -0,0 +1,36 @@
+
+extern crate ndarray;
+#[macro_use]
+extern crate ndarray_linalg;
+extern crate num_traits;
+
+use ndarray::*;
+use ndarray_linalg::*;
+
+#[test]
+fn solveh_random() {
+    let a: Array2<f64> = random_hpd(3);
+    let x: Array1<f64> = random(3);
+    let b = a.dot(&x);
+    let y = a.solveh_into(b).unwrap();
+    assert_close_l2!(&x, &y, 1e-7);
+
+    let b = a.dot(&x);
+    let f = a.factorizeh_into().unwrap();
+    let y = f.solveh_into(b).unwrap();
+    assert_close_l2!(&x, &y, 1e-7);
+}
+
+#[test]
+fn solveh_random_t() {
+    let a: Array2<f64> = random_hpd(3).reversed_axes();
+    let x: Array1<f64> = random(3);
+    let b = a.dot(&x);
+    let y = a.solveh_into(b).unwrap();
+    assert_close_l2!(&x, &y, 1e-7);
+
+    let b = a.dot(&x);
+    let f = a.factorizeh_into().unwrap();
+    let y = f.solveh_into(b).unwrap();
+    assert_close_l2!(&x, &y, 1e-7);
+}