error if not an epidf in epirecipe

NAMESPACE

@@ -272,6 +272,7 @@ importFrom(rlang,":=")
 importFrom(rlang,abort)
 importFrom(rlang,arg_match)
 importFrom(rlang,as_function)
+importFrom(rlang,caller_arg)
 importFrom(rlang,caller_env)
 importFrom(rlang,enquo)
 importFrom(rlang,enquos)

R/epi_recipe.R

@@ -16,15 +16,10 @@ epi_recipe <- function(x, ...) {
 #' @rdname epi_recipe
 #' @export
 epi_recipe.default <- function(x, ...) {
-  ## if not a formula or an epi_df, we just pass to recipes::recipe
-  if (is.matrix(x) || is.data.frame(x) || tibble::is_tibble(x)) {
-    x <- x[1, , drop = FALSE]
-  }
-  cli_warn(
-    "epi_recipe has been called with a non-epi_df object, returning a regular recipe. Various
-    step_epi_* functions will not work."
-  )
-  recipes::recipe(x, ...)
+  cli_abort(paste(
+    "`x` must be an {.cls epi_df} or a {.cls formula},",
+    "not a {.cls {class(x)[[1]]}}."
+  ))
 }
 
 #' @rdname epi_recipe
@@ -153,16 +148,15 @@ epi_recipe.formula <- function(formula, data, ...) {
   data <- data[1, ]
   # check for minus:
   if (!epiprocess::is_epi_df(data)) {
-    cli_warn(
-      "epi_recipe has been called with a non-epi_df object, returning a regular recipe. Various
-    step_epi_* functions will not work."
-    )
-    return(recipes::recipe(formula, data, ...))
+    cli_abort(paste(
+      "`epi_recipe()` has been called with a non-{.cls epi_df} object.",
+      "Use `recipe()` instead."
+    ))
   }
 
   f_funcs <- recipes:::fun_calls(formula, data)
   if (any(f_funcs == "-")) {
-    abort("`-` is not allowed in a recipe formula. Use `step_rm()` instead.")
+    cli_abort("`-` is not allowed in a recipe formula. Use `step_rm()` instead.")
   }
 
   # Check for other in-line functions

R/epipredict-package.R

@@ -1,15 +1,16 @@
 ## usethis namespace: start
 #' @importFrom tibble as_tibble
-#' @importFrom rlang := !! %||% as_function global_env set_names !!!
-#'             is_logical is_true inject enquo enquos expr sym arg_match
+#' @importFrom rlang := !! %||% as_function global_env set_names !!! caller_arg
+#' @importFrom rlang is_logical is_true inject enquo enquos expr sym arg_match
 #' @importFrom stats poly predict lm residuals quantile
 #' @importFrom dplyr arrange across all_of any_of bind_cols bind_rows group_by
-#'             summarize filter mutate select left_join rename ungroup full_join
-#'             relocate summarise everything
+#' @importFrom dplyr summarize filter mutate select left_join rename ungroup
+#' @importFrom dplyr full_join relocate summarise everything
 #' @importFrom cli cli_abort cli_warn
 #' @importFrom checkmate assert assert_character assert_int assert_scalar
-#'             assert_logical assert_numeric assert_number assert_integer
-#'             assert_integerish assert_date assert_function assert_class
+#' @importFrom checkmate assert_logical assert_numeric assert_number
+#' @importFrom checkmate assert_integer assert_integerish
+#' @importFrom checkmate assert_date assert_function assert_class
 #' @import epiprocess parsnip
 ## usethis namespace: end
 NULL

tests/testthat/_snaps/epi_recipe.md

@@ -0,0 +1,24 @@
+# epi_recipe produces error if not an epi_df
+
+    Code
+      epi_recipe(tib)
+    Condition
+      Error in `epi_recipe()`:
+      ! `x` must be an <epi_df> or a <formula>, not a <tbl_df>.
+
+---
+
+    Code
+      epi_recipe(y ~ x, tib)
+    Condition
+      Error in `epi_recipe()`:
+      ! `epi_recipe()` has been called with a non-<epi_df> object. Use `recipe()` instead.
+
+---
+
+    Code
+      epi_recipe(m)
+    Condition
+      Error in `epi_recipe()`:
+      ! `x` must be an <epi_df> or a <formula>, not a <matrix>.
+

tests/testthat/_snaps/epi_workflow.md

@@ -1,6 +1,10 @@
 # fit method does not silently drop the class
 
-    epi_recipe has been called with a non-epi_df object, returning a regular recipe. Various step_epi_* functions will not work.
+    Code
+      epi_recipe(y ~ x, data = tbl)
+    Condition
+      Error in `epi_recipe()`:
+      ! `epi_recipe()` has been called with a non-<epi_df> object. Use `recipe()` instead.
 
 ---
 

tests/testthat/test-epi_recipe.R

@@ -1,27 +1,12 @@
-test_that("epi_recipe produces default recipe", {
-  # these all call recipes::recipe(), but the template will always have 1 row
+test_that("epi_recipe produces error if not an epi_df", {
   tib <- tibble(
     x = 1:5, y = 1:5,
     time_value = seq(as.Date("2020-01-01"), by = 1, length.out = 5)
   )
-  expected_rec <- recipes::recipe(tib)
-  expected_rec$template <- expected_rec$template[1, ]
-  expect_warning(rec <- epi_recipe(tib), regexp = "epi_recipe has been called with a non-epi_df object")
-  expect_identical(expected_rec, rec)
-  expect_equal(nrow(rec$template), 1L)
-
-  expected_rec <- recipes::recipe(y ~ x, tib)
-  expected_rec$template <- expected_rec$template[1, ]
-  expect_warning(rec <- epi_recipe(y ~ x, tib), regexp = "epi_recipe has been called with a non-epi_df object")
-  expect_identical(expected_rec, rec)
-  expect_equal(nrow(rec$template), 1L)
-
+  expect_snapshot(error = TRUE, epi_recipe(tib))
+  expect_snapshot(error = TRUE, epi_recipe(y ~ x, tib))
   m <- as.matrix(tib)
-  expected_rec <- recipes::recipe(m)
-  expected_rec$template <- expected_rec$template[1, ]
-  expect_warning(rec <- epi_recipe(m), regexp = "epi_recipe has been called with a non-epi_df object")
-  expect_identical(expected_rec, rec)
-  expect_equal(nrow(rec$template), 1L)
+  expect_snapshot(error = TRUE, epi_recipe(m))
 })
 
 test_that("epi_recipe formula works", {

tests/testthat/test-epi_workflow.R

@@ -120,7 +120,7 @@ test_that("fit method does not silently drop the class", {
 
   rec_tbl <- recipe(y ~ x, data = tbl)
   rec_edf <- recipe(y ~ x, data = edf)
-  expect_snapshot_warning(erec_tbl <- epi_recipe(y ~ x, data = tbl))
+  expect_snapshot(error = TRUE, epi_recipe(y ~ x, data = tbl))
   erec_edf <- epi_recipe(y ~ x, data = edf)
 
   ewf_rec_tbl <- epi_workflow(rec_tbl, linear_reg())