Add test for noramlizers

Author: termoshtt

src/util.rs

@@ -6,6 +6,18 @@ use num_traits::Float;
 use super::vector::*;
 use std::ops::Div;
 
+/// construct matrix from diag
+pub fn from_diag<A>(d: &[A]) -> Array2<A>
+    where A: LinalgScalar
+{
+    let n = d.len();
+    let mut e = Array::zeros((n, n));
+    for i in 0..n {
+        e[(i, i)] = d[i];
+    }
+    e
+}
+
 /// stack vectors into matrix horizontally
 pub fn hstack<A, S>(xs: &[ArrayBase<S, Ix1>]) -> Result<Array<A, Ix2>, ShapeError>
     where A: LinalgScalar,

tests/normalize.rs

@@ -0,0 +1,15 @@
+include!("header.rs");
+
+#[test]
+fn n_columns() {
+    let a = random_owned(3, 2, true);
+    let (n, v) = normalize_columns(&a);
+    all_close_l2(&n.dot(&from_diag(&v)), &a, 1e-7).unwrap();
+}
+
+#[test]
+fn n_rows() {
+    let a = random_owned(3, 2, true);
+    let (v, n) = normalize_rows(&a);
+    all_close_l2(&from_diag(&v).dot(&n), &a, 1e-7).unwrap();
+}