Split simulated data up across many stages

epinowcast · May 16, 2024 · d68c3c6 · d68c3c6
1 parent ff1ce29
commit d68c3c6
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tests/testthat/setup.R b/tests/testthat/setup.R
@@ -5,11 +5,14 @@ sdlog <- 0.5
 obs_time <- 25
 sample_size <- 200
 
-sim_obs <- simulate_gillespie() |>
+obs <- simulate_gillespie() |>
   simulate_secondary(
     meanlog = meanlog,
     sdlog = sdlog
-  ) |>
-  observe_process() |>
-  filter_obs_by_obs_time(obs_time = obs_time) %>%
-  .[sample(seq_len(.N), sample_size, replace = FALSE)]
+  )
+
+obs_cens <- observe_process(obs)
+
+obs_cens_trunc <- filter_obs_by_stime(obs_cens, obs_time = obs_time)
+
+obs_cens_trunc_samp <- obs_cens_trunc[sample(seq_len(.N), sample_size, replace = FALSE)]