add rand([X], Sparse..., p::AbstractFloat, dims...)

Author: rfourquet

src/containers.jl

@@ -60,7 +60,13 @@ macro make_container(margs...)
 end
 
 @make_container(::Type{String}, [n::Integer])
+
 # sparse vectors & matrices
+@make_container(::Type{SparseVector},    p::AbstractFloat, m::Integer)
+@make_container(::Type{SparseVector},    p::AbstractFloat, d::Dims{1})
+@make_container(::Type{SparseMatrixCSC}, p::AbstractFloat, m::Integer, n::Integer)
+@make_container(::Type{SparseMatrixCSC}, p::AbstractFloat, d::Dims{2})
+
 # Tuple as a container
 @make_container(T::Type{<:Tuple})
 @make_container(::Type{Tuple}, n::Integer)

test/runtests.jl

@@ -138,21 +138,33 @@ const spString = Sampler(MersenneTwister, String)
     # sparse
     @test rand(rng..., Float64, .5, 10) isa SparseVector{Float64}
     @test rand(rng..., Float64, .5, (10,)) isa SparseVector{Float64}
+    @test rand(rng..., Float64, SparseVector, .5, 10) isa SparseVector{Float64}
+    @test rand(rng..., Float64, SparseVector, .5, (10,)) isa SparseVector{Float64}
 
     @test rand(rng..., .5, 10) isa SparseVector{Float64}
     @test rand(rng..., .5, (10,)) isa SparseVector{Float64}
+    @test rand(rng..., SparseVector, .5, 10) isa SparseVector{Float64}
+    @test rand(rng..., SparseVector, .5, (10,)) isa SparseVector{Float64}
 
     @test rand(rng..., Int, .5, 10) isa SparseVector{Int}
     @test rand(rng..., Int, .5, (10,)) isa SparseVector{Int}
+    @test rand(rng..., Int, SparseVector, .5, 10) isa SparseVector{Int}
+    @test rand(rng..., Int, SparseVector, .5, (10,)) isa SparseVector{Int}
 
     @test rand(rng..., Float64, .5, 10, 3) isa SparseMatrixCSC{Float64}
     @test rand(rng..., Float64, .5, (10, 3)) isa SparseMatrixCSC{Float64}
+    @test rand(rng..., Float64, SparseMatrixCSC, .5, 10, 3) isa SparseMatrixCSC{Float64}
+    @test rand(rng..., Float64, SparseMatrixCSC, .5, (10, 3)) isa SparseMatrixCSC{Float64}
 
     @test rand(rng..., .5, 10, 3) isa SparseMatrixCSC{Float64}
     @test rand(rng..., .5, (10, 3)) isa SparseMatrixCSC{Float64}
+    @test rand(rng..., SparseMatrixCSC, .5, 10, 3) isa SparseMatrixCSC{Float64}
+    @test rand(rng..., SparseMatrixCSC, .5, (10, 3)) isa SparseMatrixCSC{Float64}
 
     @test rand(rng..., Int, .5, 10, 3) isa SparseMatrixCSC{Int}
     @test rand(rng..., Int, .5, (10, 3)) isa SparseMatrixCSC{Int}
+    @test rand(rng..., Int, SparseMatrixCSC, .5, 10, 3) isa SparseMatrixCSC{Int}
+    @test rand(rng..., Int, SparseMatrixCSC, .5, (10, 3)) isa SparseMatrixCSC{Int}
 
     # BitArray
     for S = ([], [Bool], [Bernoulli()])