alter frosting test to avoid get_test_data bug

Author: dajmcdon

R/get_test_data.R

@@ -49,8 +49,9 @@ get_test_data <- function(recipe, x){
         .fns = ~ !is.na(.x)
       )
     ) %>%
-    dplyr::group_by(geo_value) %>%
-    dplyr::slice_tail(n = max(max_lags) + 1)
+    epiprocess::group_by(geo_value) %>%
+    dplyr::slice_tail(n = max(max_lags) + 1) %>%
+    epiprocess::ungroup()
 
   return(test_data)
 }

tests/testthat/test-frosting.R

@@ -25,7 +25,7 @@ test_that("prediction works without any postprocessor", {
     step_naomit(all_predictors()) %>%
     step_naomit(all_outcomes(), skip = TRUE)
   wf <- epi_workflow(r, linear_reg()) %>% fit(jhu)
-  latest <- get_test_data(r, jhu)
+  latest <- jhu %>% filter(time_value >= max(time_value) - 14)
 
   expect_silent(predict(wf, latest))
   p <- predict(wf, latest) %>% dplyr::filter(!is.na(.pred))