Add test for new norms

Author: termoshtt

tests/opnorm.rs

@@ -1,40 +1,27 @@
 include!("header.rs");
 
-fn assert_almost_eq(a: f64, b: f64) {
-    let rel_dev = (a - b).abs() / (a.abs() + b.abs());
-    if rel_dev > 1.0e-7 {
-        panic!("a={:?}, b={:?} are not almost equal", a, b);
-    }
-}
-
-#[test]
-fn vector_norm() {
-    let a = Array::range(1., 10., 1.);
-    assert_almost_eq(a.norm(), 285.0.sqrt());
-}
-
 #[test]
 fn matrix_opnorm_square() {
     let a = Array::range(1., 10., 1.).into_shape((3, 3)).unwrap();
-    assert_almost_eq(a.opnorm_1(), 18.0);
-    assert_almost_eq(a.opnorm_i(), 24.0);
-    assert_almost_eq(a.opnorm_f(), 285.0.sqrt());
+    a.opnorm_1().assert_close(18.0, 1e-7);
+    a.opnorm_i().assert_close(24.0, 1e-7);
+    a.opnorm_f().assert_close(285.0.sqrt(), 1e-7);
 }
 
 #[test]
 fn matrix_opnorm_square_t() {
     let a = Array::range(1., 10., 1.).into_shape((3, 3)).unwrap().reversed_axes();
-    assert_almost_eq(a.opnorm_1(), 24.0);
-    assert_almost_eq(a.opnorm_i(), 18.0);
-    assert_almost_eq(a.opnorm_f(), 285.0.sqrt());
+    a.opnorm_1().assert_close(24.0, 1e-7);
+    a.opnorm_i().assert_close(18.0, 1e-7);
+    a.opnorm_f().assert_close(285.0.sqrt(), 1e-7);
 }
 
 #[test]
 fn matrix_opnorm_3x4() {
     let a = Array::range(1., 13., 1.).into_shape((3, 4)).unwrap();
-    assert_almost_eq(a.opnorm_1(), 24.0);
-    assert_almost_eq(a.opnorm_i(), 42.0);
-    assert_almost_eq(a.opnorm_f(), 650.0.sqrt());
+    a.opnorm_1().assert_close(24.0, 1e-7);
+    a.opnorm_i().assert_close(42.0, 1e-7);
+    a.opnorm_f().assert_close(650.0.sqrt(), 1e-7);
 }
 
 #[test]
@@ -43,17 +30,17 @@ fn matrix_opnorm_3x4_t() {
         .into_shape((3, 4))
         .unwrap()
         .reversed_axes();
-    assert_almost_eq(a.opnorm_1(), 42.0);
-    assert_almost_eq(a.opnorm_i(), 24.0);
-    assert_almost_eq(a.opnorm_f(), 650.0.sqrt());
+    a.opnorm_1().assert_close(42.0, 1e-7);
+    a.opnorm_i().assert_close(24.0, 1e-7);
+    a.opnorm_f().assert_close(650.0.sqrt(), 1e-7);
 }
 
 #[test]
 fn matrix_opnorm_4x3() {
     let a = Array::range(1., 13., 1.).into_shape((4, 3)).unwrap();
-    assert_almost_eq(a.opnorm_1(), 30.0);
-    assert_almost_eq(a.opnorm_i(), 33.0);
-    assert_almost_eq(a.opnorm_f(), 650.0.sqrt());
+    a.opnorm_1().assert_close(30.0, 1e-7);
+    a.opnorm_i().assert_close(33.0, 1e-7);
+    a.opnorm_f().assert_close(650.0.sqrt(), 1e-7);
 }
 
 #[test]
@@ -62,7 +49,7 @@ fn matrix_opnorm_4x3_t() {
         .into_shape((4, 3))
         .unwrap()
         .reversed_axes();
-    assert_almost_eq(a.opnorm_1(), 33.0);
-    assert_almost_eq(a.opnorm_i(), 30.0);
-    assert_almost_eq(a.opnorm_f(), 650.0.sqrt());
+    a.opnorm_1().assert_close(33.0, 1e-7);
+    a.opnorm_i().assert_close(30.0, 1e-7);
+    a.opnorm_f().assert_close(650.0.sqrt(), 1e-7);
 }

tests/vector.rs

@@ -0,0 +1,23 @@
+include!("header.rs");
+
+#[test]
+fn vector_norm() {
+    let a = Array::range(1., 10., 1.);
+    a.norm().assert_close(285.0.sqrt(), 1e-7);
+}
+
+#[test]
+fn vector_norm_l1() {
+    let a = arr1(&[1.0, -1.0]);
+    a.norm_l1().assert_close(2.0, 1e-7);
+    let b = arr2(&[[0.0, -1.0], [1.0, 0.0]]);
+    b.norm_l1().assert_close(2.0, 1e-7);
+}
+
+#[test]
+fn vector_norm_max() {
+    let a = arr1(&[1.0, 1.0, -3.0]);
+    a.norm_max().assert_close(3.0, 1e-7);
+    let b = arr2(&[[1.0, 3.0], [1.0, -4.0]]);
+    b.norm_max().assert_close(4.0, 1e-7);
+}