From af8c394f1083841a38565da0fd702f6207ed8565 Mon Sep 17 00:00:00 2001
From: Samuel Brand <sam055@mac.com>
Date: Sun, 6 Oct 2024 15:45:32 +0100
Subject: [PATCH] update tests

---
 pipeline/test/pipeline/test_pipelinefunctions.jl | 12 ++++++++++++
 pipeline/test/plotting/prior_pred_plot.jl        |  5 ++---
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/pipeline/test/pipeline/test_pipelinefunctions.jl b/pipeline/test/pipeline/test_pipelinefunctions.jl
index 0a49944f3..b9836a6bf 100644
--- 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
index eb0d43234..8c66687fb 100644
--- 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