Move test into tests/ dir

Author: termoshtt

src/krylov/arnoldi.rs

@@ -130,39 +130,3 @@ where
     let mgs = MGS::new(v.len(), tol);
     Arnoldi::new(mul_mat(a), v, mgs).complete()
 }
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-    use crate::{assert::*, generate::*};
-
-    #[test]
-    fn aq_qh_mgs() {
-        let a: Array2<f64> = random((5, 5));
-        let v: Array1<f64> = random(5);
-        let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
-        println!("A = \n{:?}", &a);
-        println!("Q = \n{:?}", &q);
-        println!("H = \n{:?}", &h);
-        let aq = a.dot(&q);
-        let qh = q.dot(&h);
-        println!("AQ = \n{:?}", &aq);
-        println!("QH = \n{:?}", &qh);
-        close_l2(&aq, &qh, 1e-9);
-    }
-
-    #[test]
-    fn aq_qh_householder() {
-        let a: Array2<f64> = random((5, 5));
-        let v: Array1<f64> = random(5);
-        let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
-        println!("A = \n{:?}", &a);
-        println!("Q = \n{:?}", &q);
-        println!("H = \n{:?}", &h);
-        let aq = a.dot(&q);
-        let qh = q.dot(&h);
-        println!("AQ = \n{:?}", &aq);
-        println!("QH = \n{:?}", &qh);
-        close_l2(&aq, &qh, 1e-9);
-    }
-}

src/krylov/mod.rs

@@ -7,6 +7,7 @@ pub mod arnoldi;
 pub mod householder;
 pub mod mgs;
 
+pub use arnoldi::{arnoldi_householder, arnoldi_mgs, Arnoldi};
 pub use householder::{householder, Householder};
 pub use mgs::{mgs, MGS};
 

tests/arnoldi.rs

@@ -0,0 +1,62 @@
+use ndarray::*;
+use ndarray_linalg::{krylov::*, *};
+
+#[test]
+fn aq_qh_mgs() {
+    let a: Array2<f64> = random((5, 5));
+    let v: Array1<f64> = random(5);
+    let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
+    println!("A = \n{:?}", &a);
+    println!("Q = \n{:?}", &q);
+    println!("H = \n{:?}", &h);
+    let aq = a.dot(&q);
+    let qh = q.dot(&h);
+    println!("AQ = \n{:?}", &aq);
+    println!("QH = \n{:?}", &qh);
+    close_l2(&aq, &qh, 1e-9);
+}
+
+#[test]
+fn aq_qh_householder() {
+    let a: Array2<f64> = random((5, 5));
+    let v: Array1<f64> = random(5);
+    let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
+    println!("A = \n{:?}", &a);
+    println!("Q = \n{:?}", &q);
+    println!("H = \n{:?}", &h);
+    let aq = a.dot(&q);
+    let qh = q.dot(&h);
+    println!("AQ = \n{:?}", &aq);
+    println!("QH = \n{:?}", &qh);
+    close_l2(&aq, &qh, 1e-9);
+}
+
+#[test]
+fn aq_qh_mgs_complex() {
+    let a: Array2<c64> = random((5, 5));
+    let v: Array1<c64> = random(5);
+    let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
+    println!("A = \n{:?}", &a);
+    println!("Q = \n{:?}", &q);
+    println!("H = \n{:?}", &h);
+    let aq = a.dot(&q);
+    let qh = q.dot(&h);
+    println!("AQ = \n{:?}", &aq);
+    println!("QH = \n{:?}", &qh);
+    close_l2(&aq, &qh, 1e-9);
+}
+
+#[test]
+fn aq_qh_householder_complex() {
+    let a: Array2<c64> = random((5, 5));
+    let v: Array1<c64> = random(5);
+    let (q, h) = arnoldi_mgs(a.clone(), v, 1e-9);
+    println!("A = \n{:?}", &a);
+    println!("Q = \n{:?}", &q);
+    println!("H = \n{:?}", &h);
+    let aq = a.dot(&q);
+    let qh = q.dot(&h);
+    println!("AQ = \n{:?}", &aq);
+    println!("QH = \n{:?}", &qh);
+    close_l2(&aq, &qh, 1e-9);
+}