Add test for SVD without u or vt

Author: termoshtt

tests/svd.rs

@@ -3,6 +3,12 @@ use ndarray_linalg::*;
 use std::cmp::min;
 
 fn test(a: &Array2<f64>, n: usize, m: usize) {
+    test_both(a, n, m);
+    test_u(a, n, m);
+    test_vt(a, n, m);
+}
+
+fn test_both(a: &Array2<f64>, n: usize, m: usize) {
     let answer = a.clone();
     println!("a = \n{:?}", a);
     let (u, s, vt): (_, Array1<_>, _) = a.svd(true, true).unwrap();
@@ -18,6 +24,26 @@ fn test(a: &Array2<f64>, n: usize, m: usize) {
     assert_close_l2!(&u.dot(&sm).dot(&vt), &answer, 1e-7);
 }
 
+fn test_u(a: &Array2<f64>, n: usize, _m: usize) {
+    println!("a = \n{:?}", a);
+    let (u, _s, vt): (_, Array1<_>, _) = a.svd(true, false).unwrap();
+    assert!(u.is_some());
+    assert!(vt.is_none());
+    let u = u.unwrap();
+    assert_eq!(u.dim().0, n);
+    assert_eq!(u.dim().1, n);
+}
+
+fn test_vt(a: &Array2<f64>, _n: usize, m: usize) {
+    println!("a = \n{:?}", a);
+    let (u, _s, vt): (_, Array1<_>, _) = a.svd(false, true).unwrap();
+    assert!(u.is_none());
+    assert!(vt.is_some());
+    let vt = vt.unwrap();
+    assert_eq!(vt.dim().0, m);
+    assert_eq!(vt.dim().1, m);
+}
+
 #[test]
 fn svd_square() {
     let a = random((3, 3));