unit tests for tidymodels/parsnip#855

Author: topepo

DESCRIPTION

@@ -40,6 +40,7 @@ Suggests:
     pec,
     pillar (>= 1.6.5.9001),
     poissonreg,
+    prodlim,
     pscl,
     purrr,
     randomForest,
@@ -54,6 +55,7 @@ Suggests:
     spatialsample,
     spelling,
     stacks,
+    survival,
     testthat,
     themis,
     tibble,
@@ -68,9 +70,9 @@ Remotes:
     r-lib/pillar
 biocViews: mixOmics
 Config/testthat/edition: 3
+Config/testthat/start-first: parsnip-extension-messaging
 Encoding: UTF-8
 Language: en-US
 LazyData: true
 Roxygen: list(markdown = TRUE)
 RoxygenNote: 7.1.1.9001
-Config/testthat/start-first: parsnip-extension-messaging

tests/testthat/test-censor-probs.R

@@ -0,0 +1,47 @@
+
+test_that("reverse Kaplan-Meier curves", {
+  skip_if_not_installed("prodlim")
+  skip_if_not_installed("survival")
+  skip_if_not_installed("censored")
+  skip_if(utils::packageVersion("parsnip") < "1.0.3.9001")
+
+  library(survival)
+  library(censored)
+
+  lung <- lung[complete.cases(lung), ]
+
+  mod_fit <-
+    survival_reg() %>%
+    fit(Surv(time, status) ~ age + sex, data = lung)
+
+  expect_true(any(names(mod_fit) == "censor_probs"))
+  expect_true(
+    inherits(mod_fit$censor_probs,
+             c("censoring_model_reverse_km", "censoring_model")))
+  expect_equal(
+    names(mod_fit$censor_probs),
+    c("formula", "fit", "label", "required_pkgs")
+  )
+  expect_true(inherits(mod_fit$censor_probs$fit, "prodlim"))
+  expect_equal(
+    mod_fit$censor_probs$formula,
+    Surv(time, status) ~ age + sex,
+    ignore_formula_env = TRUE
+  )
+  expect_equal(
+    mod_fit$censor_probs$fit$formula,
+    Surv(time, status) ~ 1,
+    ignore_formula_env = TRUE
+  )
+  expect_equal(
+    environment(mod_fit$censor_probs$formula),
+    rlang::base_env()
+  )
+  expect_equal(
+    environment(mod_fit$censor_probs$fit$formula),
+    rlang::base_env()
+  )
+  expect_equal(mod_fit$censor_probs$label, "reverse_km")
+  expect_equal(mod_fit$censor_probs$required_pkgs, "prodlim")
+  expect_snapshot(print(mod_fit$censor_probs))
+})