Test for Cholesky

Author: termoshtt

tests/cholesky.rs

@@ -1,25 +1,30 @@
 include!("header.rs");
 
-#[test]
-fn cholesky() {
-    let r_dist = RealNormal::new(0., 1.);
-    let mut a = Array::<f64, _>::random((3, 3), r_dist);
-    a = a.dot(&a.t());
-    println!("a = \n{:?}", a);
-    let c = a.clone().cholesky().unwrap();
-    println!("c = \n{:?}", c);
-    println!("cc = \n{:?}", c.t().dot(&c));
-    all_close_l2(&c.t().dot(&c), &a, 1e-7).unwrap();
+macro_rules! impl_test {
+    ($modname:ident, $clone:ident) => {
+mod $modname {
+    use super::random_hermite;
+    use ndarray_linalg::prelude::*;
+    #[test]
+    fn cholesky() {
+        let a = random_hermite(3);
+        println!("a = \n{:?}", a);
+        let c = a.$clone().cholesky().unwrap();
+        println!("c = \n{:?}", c);
+        println!("cc = \n{:?}", c.t().dot(&c));
+        all_close_l2(&c.t().dot(&c), &a, 1e-7).unwrap();
+    }
+    #[test]
+    fn cholesky_t() {
+        let a = random_hermite(3);
+        println!("a = \n{:?}", a);
+        let c = a.$clone().cholesky().unwrap();
+        println!("c = \n{:?}", c);
+        println!("cc = \n{:?}", c.t().dot(&c));
+        all_close_l2(&c.t().dot(&c), &a, 1e-7).unwrap();
+    }
 }
+}} // impl_test
 
-#[test]
-fn cholesky_t() {
-    let r_dist = RealNormal::new(0., 1.);
-    let mut a = Array::<f64, _>::random((3, 3), r_dist);
-    a = a.dot(&a.t()).reversed_axes();
-    println!("a = \n{:?}", a);
-    let c = a.clone().cholesky().unwrap();
-    println!("c = \n{:?}", c);
-    println!("cc = \n{:?}", c.t().dot(&c));
-    all_close_l2(&c.t().dot(&c), &a, 1e-7).unwrap();
-}
+impl_test!(owned, clone);
+impl_test!(shared, to_shared);

tests/det.rs

@@ -1,6 +1,6 @@
 include!("header.rs");
 
-macro_rules! impl_test_det {
+macro_rules! impl_test{
     ($modname:ident, $clone:ident) => {
 mod $modname {
     use super::random_hermite;
@@ -15,7 +15,7 @@ mod $modname {
         deth.assert_close(det_eig, 1.0e-7);
     }
 }
-}} // impl_test_det
+}} // impl_test
 
-impl_test_det!(owned, clone);
-impl_test_det!(shared, to_shared);
+impl_test!(owned, clone);
+impl_test!(shared, to_shared);

tests/eigh.rs

@@ -1,6 +1,6 @@
 include!("header.rs");
 
-macro_rules! impl_test_eigh {
+macro_rules! impl_test {
     ($modname:ident, $clone:ident) => {
 mod $modname {
     use ndarray::prelude::*;
@@ -27,7 +27,7 @@ mod $modname {
         }
     }
 }
-}} // impl_test_eigh
+}} // impl_test
 
-impl_test_eigh!(owned, clone);
-impl_test_eigh!(shared, to_shared);
+impl_test!(owned, clone);
+impl_test!(shared, to_shared);