change type of some tests

CDCgov · Jun 10, 2024 · a1a100e · a1a100e
1 parent 6e4d890
commit a1a100e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/EpiAware/test/EpiAwareUtils/generate_epiware.jl b/EpiAware/test/EpiAwareUtils/generate_epiware.jl
@@ -32,7 +32,7 @@
     gen = generated_quantities(test_mdl, rand(test_mdl))
 
     #Check model sampled
-    @test eltype(gen.generated_y_t) <: Int
+    @test eltype(gen.generated_y_t) <: Union{Missing, Real}
     @test eltype(gen.I_t) <: AbstractFloat
     @test length(gen.I_t) == time_horizon
 end
@@ -70,7 +70,7 @@ end
     gens = generated_quantities(test_mdl, chn)
 
     #Check model sampled
-    @test eltype(gens[1].generated_y_t) <: Int
+    @test eltype(gens[1].generated_y_t) <: Union{Missing, Real}
     @test eltype(gens[1].I_t) <: AbstractFloat
     @test length(gens[1].I_t) == time_horizon
 end
@@ -110,7 +110,7 @@ end
     gens = generated_quantities(test_mdl, chn)
 
     #Check model sampled
-    @test eltype(gens[1].generated_y_t) <: Int
+    @test eltype(gens[1].generated_y_t) <: Union{Missing, Real}
     @test eltype(gens[1].I_t) <: AbstractFloat
     @test length(gens[1].I_t) == time_horizon
 end