fix SafePoisson to be Discrete dist

CDCgov · Aug 20, 2024 · 8ed5a8d · 8ed5a8d
1 parent 8c383d2
commit 8ed5a8d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/EpiAware/src/EpiAwareUtils/SafePoisson.jl b/EpiAware/src/EpiAwareUtils/SafePoisson.jl
@@ -45,7 +45,7 @@ var(d)
 7.016735912097631e20
 ```
 "
-struct SafePoisson{T <: Real} <: ContinuousUnivariateDistribution
+struct SafePoisson{T <: Real} <: DiscreteUnivariateDistribution
     λ::T
 
     SafePoisson{T}(λ::Real) where {T <: Real} = new{T}(λ)

diff --git a/EpiAware/test/EpiAwareUtils/SafePoisson.jl b/EpiAware/test/EpiAwareUtils/SafePoisson.jl
@@ -2,6 +2,9 @@
     λ = 10.0
     dist = SafePoisson(λ)
     @test typeof(dist) <: SafePoisson
+    @test rand(dist) isa Int
+    @test rand(dist, 10) isa Vector{Int}
+    @test rand(dist, 10, 10) isa Array{Int}
 end
 
 @testitem "Check distribution properties of SafePoisson" begin