Two missing tests for sprand (#30551)

JuliaLang · Jan 2, 2019 · 2c64220 · 2c64220
1 parent c154215
commit 2c64220
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/stdlib/SparseArrays/test/sparsevector.jl b/stdlib/SparseArrays/test/sparsevector.jl
@@ -170,6 +170,22 @@ end
             @test sprandn(r1, 100, .9) == sprandn(r2, 100, .9)
             @test sprand(r1, Bool, 100, .9) == sprand(r2,  Bool, 100, .9)
         end
+
+        # test sprand with function inputs
+        let xr = sprand(1000, 0.9, rand)
+            @test isa(xr, SparseVector{Float64,Int})
+            @test length(xr) == 1000
+            if !isempty(nonzeros(xr))
+                @test all(nonzeros(xr) .> 0.0)
+            end
+        end
+        let xr = sprand(1000, 0.9, rand, Float32)
+            @test isa(xr, SparseVector{Float32,Int})
+            @test length(xr) == 1000
+            if !isempty(nonzeros(xr))
+                @test all(nonzeros(xr) .> 0.0)
+            end
+        end
     end
 end
 ### Element access