fix FixedIntercept test

CDCgov · Jul 10, 2024 · 6efa1f0 · 6efa1f0
1 parent 1dc6fce
commit 6efa1f0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/EpiAware/test/EpiLatentModels/models/FixedIntercept.jl b/EpiAware/test/EpiLatentModels/models/FixedIntercept.jl
@@ -10,6 +10,8 @@ end
     int = FixedIntercept(0.1)
     int_model = generate_latent(int, 10)
     int_model_out = int_model()
-    @test length(int_model_out[1]) == 10
-    @test all(x -> x == int_model_out[2].intercept, int_model_out[1])
+    rand_model = rand(int_model)
+    @test rand_model == NamedTuple()
+    @test length(int_model_out) == 10
+    @test all(x -> x == 0.1, int_model_out)
 end