Test for inv()

Author: termoshtt

tests/header.rs

@@ -16,6 +16,11 @@ use ndarray_rand::RandomExt;
 #[allow(unused_imports)]
 use num_traits::Float;
 
+pub fn random_square(n: usize) -> Array<f64, Ix2> {
+    let r_dist = RealNormal::new(0., 1.);
+    Array::<f64, _>::random((n, n), r_dist)
+}
+
 pub fn random_hermite(n: usize) -> Array<f64, Ix2> {
     let r_dist = RealNormal::new(0., 1.);
     let a = Array::<f64, _>::random((n, n), r_dist);

tests/inv.rs

@@ -1,27 +1,36 @@
 include!("header.rs");
 
-#[test]
-fn inv_random() {
-    let r_dist = RealNormal::new(0., 1.);
-    let a = Array::<f64, _>::random((3, 3), r_dist);
-    let ai = a.clone().inv().unwrap();
-    let id = Array::eye(3);
-    all_close_l2(&ai.dot(&a), &id, 1e-7).unwrap();
-}
+macro_rules! impl_test{
+    ($modname:ident, $clone:ident) => {
+mod $modname {
+    use super::random_square;
+    use ndarray::prelude::*;
+    use ndarray_linalg::prelude::*;
+    #[test]
+    fn inv_random() {
+        let a = random_square(3);
+        let ai = a.$clone().inv().unwrap();
+        let id = Array::eye(3);
+        all_close_l2(&ai.dot(&a), &id, 1e-7).unwrap();
+    }
 
-#[test]
-fn inv_random_t() {
-    let r_dist = RealNormal::new(0., 1.);
-    let a = Array::<f64, _>::random((3, 3), r_dist).reversed_axes();
-    let ai = a.clone().inv().unwrap();
-    let id = Array::eye(3);
-    all_close_l2(&ai.dot(&a), &id, 1e-7).unwrap();
-}
+    #[test]
+    fn inv_random_t() {
+        let a = random_square(3).reversed_axes();
+        let ai = a.$clone().inv().unwrap();
+        let id = Array::eye(3);
+        all_close_l2(&ai.dot(&a), &id, 1e-7).unwrap();
+    }
 
-#[test]
-#[should_panic]
-fn inv_error() {
-    // do not have inverse
-    let a = Array::range(1., 10., 1.).into_shape((3, 3)).unwrap();
-    let _ = a.clone().inv().unwrap();
+    #[test]
+    #[should_panic]
+    fn inv_error() {
+        // do not have inverse
+        let a = Array::range(1., 10., 1.).into_shape((3, 3)).unwrap();
+        let _ = a.$clone().inv().unwrap();
+    }
 }
+}} // impl_test
+
+impl_test!(owned, clone);
+impl_test!(shared, to_shared);