update tests

CDCgov · Oct 6, 2024 · af8c394 · af8c394
1 parent d733d90
commit af8c394
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/pipeline/test/pipeline/test_pipelinefunctions.jl b/pipeline/test/pipeline/test_pipelinefunctions.jl
@@ -43,3 +43,15 @@ end
     fetch(res)
     @test isnothing(res)
 end
+
+@testset "do_pipeline test: prior predictive" begin
+    using Dagger
+    pipelines = map([SmoothOutbreakPipeline, MeasuresOutbreakPipeline,
+        SmoothEndemicPipeline, RoughEndemicPipeline]) do pipetype
+        pipetype(; ndraws = 10, nchains = 1, testmode = true, priorpredictive = true)
+    end
+
+    res = do_pipeline(pipelines)
+    fetch(res)
+    @test isnothing(res)
+end
diff --git a/pipeline/test/plotting/prior_pred_plot.jl b/pipeline/test/plotting/prior_pred_plot.jl
@@ -5,14 +5,13 @@
         SmoothEndemicPipeline, RoughEndemicPipeline] |> rand
     P = pipetype(; testmode = true, nchains = 1, ndraws = 2000, priorpredictive = true)
     inference_config = make_inference_configs(P) |> rand
-
+    inference_config["T"] = 21
     #Add missing data
     missingdata = Dict("y_t" => missing, "I_t" => fill(1.0, 100), "truth_I0" => 1.0,
         "truth_gi_mean" => inference_config["gi_mean"])
     results = generate_inference_results(missingdata, inference_config, P)
 
-    fig = prior_predictive_plot(results["inference_config"], results["inference_results"],
-        results["epiprob"]; ps = [0.025, 0.1, 0.25])
+    fig = results["priorpredictive"]
 
     @test fig isa Figure
 end