works in the forecaster

cmu-delphi · Oct 26, 2024 · d36760b · d36760b
1 parent f6d01e2
commit d36760b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/testthat/test-grf_quantiles.R b/tests/testthat/test-grf_quantiles.R
@@ -70,4 +70,10 @@ test_that("quantile_rand_forest operates with arx_forecaster", {
     rlang::eval_tidy(spec2$eng_args$quantiles),
     c(.05, .1, .5, .9, .95) # merged with arx_args default
   )
+  df <- epidatasets::counts_subset %>% filter(time_value >= "2021-10-01")
+  z <- arx_forecaster(df, "cases", "cases", spec)
+  expect_identical(
+    nested_quantiles(z$predictions$.pred_distn[1])[[1]]$quantile_levels,
+    c(.05, .1, .5, .9, .95)
+  )
 })