add tests for check_outcome()

Author: EmilHvitfeldt

tests/testthat/_snaps/misc.md

@@ -124,3 +124,27 @@
       Error in `set_engine()`:
       ! `set_engine()` expected a model specification to be supplied to the `object` argument, but received a(n) `data.frame` object.
 
+# check_outcome works as expected
+
+    Code
+      check_outcome(factor(1:10), reg_spec)
+    Condition
+      Error in `check_outcome()`:
+      ! For a regression model, the outcome should be `numeric`, not a `factor`.
+
+---
+
+    Code
+      check_outcome(1:10, class_spec)
+    Condition
+      Error in `check_outcome()`:
+      ! For a classification model, the outcome should be a `factor`, not a `integer`.
+
+---
+
+    Code
+      check_outcome(1:10, cens_spec)
+    Condition
+      Error in `check_outcome()`:
+      ! For a censored regression model, the outcome should be a `Surv` object, not a `integer`.
+

tests/testthat/test_misc.R

@@ -181,3 +181,48 @@ test_that('set_engine works as a generic', {
   )
 
 })
+
+test_that('check_outcome works as expected', {
+  reg_spec <- linear_reg()
+
+  expect_no_error(
+    check_outcome(1:10, reg_spec)
+  )
+
+  expect_no_error(
+    check_outcome(mtcars, reg_spec)
+  )
+
+  expect_snapshot(
+    error = TRUE,
+    check_outcome(factor(1:10), reg_spec)
+  )
+
+  class_spec <- logistic_reg()
+
+  expect_no_error(
+    check_outcome(factor(1:10), class_spec)
+  )
+
+  expect_no_error(
+    check_outcome(lapply(mtcars, as.factor), class_spec)
+  )
+
+  expect_snapshot(
+    error = TRUE,
+    check_outcome(1:10, class_spec)
+  )
+
+  # Fake specification to avoid having to load {censored}
+  cens_spec <- logistic_reg()
+  cens_spec$mode <- "censored regression"
+
+  expect_no_error(
+    check_outcome(survival::Surv(1, 1), cens_spec)
+  )
+
+  expect_snapshot(
+    error = TRUE,
+    check_outcome(1:10, cens_spec)
+  )
+})

tests/testthat/test_nearest_neighbor_kknn.R

@@ -24,7 +24,7 @@ test_that('kknn execution', {
       x = hpc[, num_pred],
       y = hpc$input_fields
     ),
-    regexp = "outcome should be a factor"
+    regexp = "outcome should be a `factor`"
   )
 
   # nominal