some initial tests

Author: ‘topepo’

R/quantiles.R

@@ -9,6 +9,8 @@ check_quantile_level <- function(x, object, call) {
       {.arg quantile_level} must be specified for quantile regression models.")
     }
   }
+  # TODO we need better vectorization here, otherwise we get things like:
+  # "Error during wrapup: i In index: 2." in the traceback.
   res <-
     purrr::map(x,
                ~ check_number_decimal(.x, min = 0, max = 1,

tests/testthat/_snaps/quantile-reg-specs.md

@@ -0,0 +1,49 @@
+# test mode setting for quantile regression
+
+    Code
+      linear_reg() %>% set_engine("quantreg") %>% set_mode("regression")
+    Condition
+      Error in `set_mode()`:
+      ! Available modes for engine quantreg are: "unknown" and "quantile regression".
+
+---
+
+    Code
+      linear_reg() %>% set_engine("quantreg") %>% set_mode("quantile regression")
+    Condition
+      Error in `check_quantile_level()`:
+      ! In `check_mode()`, at least one value of `quantile_level` must be specified for quantile regression models.
+
+---
+
+    Code
+      linear_reg() %>% set_engine("quantreg") %>% set_mode("quantile regression",
+        quantile_level = 2)
+    Condition
+      Error in `purrr::map()`:
+      i In index: 1.
+      Caused by error in `set_mode()`:
+      ! `quantile_level` must be a number between 0 and 1, not the number 2.
+
+---
+
+    Code
+      linear_reg() %>% set_engine("quantreg") %>% set_mode("quantile regression",
+        quantile_level = 1:2)
+    Condition
+      Error in `purrr::map()`:
+      i In index: 2.
+      Caused by error in `set_mode()`:
+      ! `quantile_level` must be a number between 0 and 1, not the number 2.
+
+---
+
+    Code
+      linear_reg() %>% set_engine("quantreg") %>% set_mode("quantile regression",
+        quantile_level = NA_real_)
+    Condition
+      Error in `purrr::map()`:
+      i In index: 1.
+      Caused by error in `set_mode()`:
+      ! `quantile_level` must be a number, not a numeric `NA`.
+

tests/testthat/test-quantile-reg-specs.R

@@ -0,0 +1,33 @@
+test_that('test mode setting for quantile regression', {
+
+  expect_snapshot(
+    linear_reg() %>%
+      set_engine("quantreg") %>%
+      set_mode("regression"),
+    error = TRUE)
+
+  expect_snapshot(
+    linear_reg() %>%
+      set_engine("quantreg") %>%
+      set_mode("quantile regression"),
+    error = TRUE)
+
+  expect_snapshot(
+    linear_reg() %>%
+      set_engine("quantreg") %>%
+      set_mode("quantile regression", quantile_level = 2),
+    error = TRUE)
+
+  # TODO this needs to be improved
+  expect_snapshot(
+    linear_reg() %>%
+      set_engine("quantreg") %>%
+      set_mode("quantile regression", quantile_level = 1:2),
+    error = TRUE)
+
+  expect_snapshot(
+    linear_reg() %>%
+      set_engine("quantreg") %>%
+      set_mode("quantile regression", quantile_level = NA_real_),
+    error = TRUE)
+})