Add test for squeeze on sparse matrices.

JuliaLang · Feb 26, 2016 · 7c54abd · 7c54abd
1 parent 4563526
commit 7c54abd
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/sparsedir/sparse.jl b/test/sparsedir/sparse.jl
@@ -74,6 +74,18 @@ a116[p, p] = reshape(1:9, 3, 3)
 s116[p, p] = reshape(1:9, 3, 3)
 @test a116 == s116
 
+# squeeze
+for i = 1:5
+    am = sprand(20, 1, 0.2)
+    av = squeeze(am, 2)
+    @test ndims(av) == 1
+    @test all(av.==am)
+    am = sprand(1, 20, 0.2)
+    av = squeeze(am, 1)
+    @test ndims(av) == 1
+    @test all(av.'.==am)
+end
+
 # matrix-vector multiplication (non-square)
 for i = 1:5
     a = sprand(10, 5, 0.5)