enable rand([X], ::Type{<:NamedTuple})

Author: rfourquet

README.md

@@ -180,9 +180,12 @@ julia> julia> rand(Bernoulli(0.2), BitVector, 10) # using the Bernoulli distribu
 julia> rand(1:3, NTuple{3}) # NTuple{3} considered as a container, equivalent to rand(make(NTuple{3}, 1:3))
 (3, 3, 1)
 
-julia> rand(1:3, Tuple{Int,UInt8, BigFloat}) # works also with more general tuple types
+julia> rand(1:3, Tuple{Int,UInt8, BigFloat}) # works also with more general tuple types ...
 (3, 0x02, 2.0)
 
+julia> rand(1:3, NamedTuple{(:a, :b)}) # ... and with named tuples
+(a = 3, b = 2)
+
 julia> RandomExtensions.random_staticarrays() # poor man's conditional modules!
 # ugly warning
 

src/containers.jl

@@ -75,6 +75,12 @@ end
 @make_container(T::Type{<:Tuple})
 @make_container(::Type{Tuple}, n::Integer)
 @make_container(T::Type{NTuple{N,TT} where N} => TT, n::Integer)
+@make_container(T::Type{<:NamedTuple{K}} => K)
+
+if VERSION < v"1.1.0"
+    # disambiguate
+    rand(rng::AbstractRNG, ::Type{NamedTuple}) = rand(rng, make(NamedTuple))
+end
 
 ## arrays (same as in Random, but with explicit type specification, e.g. rand(Int, Array, 4)
 

test/runtests.jl

@@ -416,6 +416,12 @@ end
     t = rand(make(NamedTuple{(:a, :b),Tuple{Float64,UInt8}}, 1:3))
     @test t isa NamedTuple{(:a, :b), Tuple{Float64,UInt8}}
     @test t.a ∈ 1:3 && t.b ∈ 1:3
+
+    # as container
+    @test rand(1:3, NamedTuple{(:a,)}) isa NamedTuple{(:a,), Tuple{Int}}
+    @test rand(1:3, NamedTuple{(:a,), Tuple{Float64}}) isa NamedTuple{(:a,), Tuple{Float64}}
+    @test rand(1:3, NamedTuple{(:a, :b)}) isa NamedTuple{(:a, :b), Tuple{Int,Int}}
+    @test rand(1:3, NamedTuple{(:a, :b), Tuple{Float64,Float64}}) isa NamedTuple{(:a, :b), Tuple{Float64,Float64}}
 end
 
 @testset "rand(make(String, ...))" begin