Define Random.Sampler

JuliaStats · Apr 26, 2021 · f4dc84d · f4dc84d
1 parent a36c613
commit f4dc84d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/genericrand.jl b/src/genericrand.jl
@@ -55,3 +55,7 @@ with this `sampler` method, which would be used for batch sampling.
 The general fallback is `sampler(d::Distribution) = d`.
 """
 sampler(s::Sampleable) = s
+
+# Random API
+Random.Sampler(::Type{<:AbstractRNG}, s::Sampleable, ::Val{1}) = s
+Random.Sampler(::Type{<:AbstractRNG}, s::Sampleable, ::Val{Inf}) = sampler(s)
diff --git a/test/samplers.jl b/test/samplers.jl
@@ -100,4 +100,20 @@ import Distributions:
             test_samples(S(d), d, n_tsamples, rng=rng)
         end
     end
+
+    @testset "Random.Sampler" begin
+        for dist in (
+            Binomial(5, 0.3),
+            Exponential(2.0),
+            Gamma(0.1, 1.0),
+            Gamma(2.0, 1.0),
+            MatrixNormal(3, 4),
+            MvNormal(3, 1.0),
+            Normal(1.5, 2.0),
+            Poisson(0.5),
+        )
+            @test Random.Sampler(rng, dist, Val(1)) == dist
+            @test Random.Sampler(rng, dist) == sampler(dist)
+        end
+    end
 end