From fdaf8c0fbe8cbdf3bfc1721a6bc1838cfa43a529 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Crozet?= <developer@crozet.re>
Date: Thu, 30 Dec 2021 23:03:22 +0100
Subject: [PATCH] Add tests for Matrix::pow

---
 tests/linalg/mod.rs |  1 +
 tests/linalg/pow.rs | 49 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 tests/linalg/pow.rs

diff --git a/tests/linalg/mod.rs b/tests/linalg/mod.rs
index 9c252bfd..d9bd6cd9 100644
--- a/tests/linalg/mod.rs
+++ b/tests/linalg/mod.rs
@@ -9,6 +9,7 @@ mod full_piv_lu;
 mod hessenberg;
 mod inverse;
 mod lu;
+mod pow;
 mod qr;
 mod schur;
 mod solve;
diff --git a/tests/linalg/pow.rs b/tests/linalg/pow.rs
new file mode 100644
index 00000000..22a1a0c0
--- /dev/null
+++ b/tests/linalg/pow.rs
@@ -0,0 +1,49 @@
+#[cfg(feature = "proptest-support")]
+mod proptest_tests {
+    macro_rules! gen_tests(
+        ($module: ident, $scalar: expr, $scalar_type: ty) => {
+            mod $module {
+                use na::DMatrix;
+                #[allow(unused_imports)]
+                use crate::core::helper::{RandScalar, RandComplex};
+                use std::cmp;
+
+                use crate::proptest::*;
+                use proptest::{prop_assert, proptest};
+
+                proptest! {
+                    #[test]
+                    fn pow(n in PROPTEST_MATRIX_DIM, p in 0u32..=4) {
+                        let n = cmp::max(1, cmp::min(n, 10));
+                        let m = DMatrix::<$scalar_type>::new_random(n, n).map(|e| e.0);
+                        let m_pow = m.pow(p);
+                        let mut expected = m.clone();
+                        expected.fill_with_identity();
+
+                        for _ in 0..p {
+                            expected = &m * &expected;
+                        }
+
+                        prop_assert!(relative_eq!(m_pow, expected, epsilon = 1.0e-5))
+                    }
+
+                    #[test]
+                    fn pow_static_square_4x4(m in matrix4_($scalar), p in 0u32..=4) {
+                        let mut expected = m.clone();
+                        let m_pow = m.pow(p);
+                        expected.fill_with_identity();
+
+                        for _ in 0..p {
+                            expected = &m * &expected;
+                        }
+
+                        prop_assert!(relative_eq!(m_pow, expected, epsilon = 1.0e-5))
+                    }
+                }
+            }
+        }
+    );
+
+    gen_tests!(complex, complex_f64(), RandComplex<f64>);
+    gen_tests!(f64, PROPTEST_F64, RandScalar<f64>);
+}