tests/testthat/test_paramtest_penalized_surv_penalized.R

In mlr-org/mlr3extralearners: Extra Learners For mlr3

test_that("paramtest surv.penalized train", {
  learner = lrn("surv.penalized")
  fun_list = list(penalized::penalized)
  exclude = c(
    "response", # coerced internally
    "penalized", # all variables assumed penalized unless passed to 'unpenalized' parameter
    "data", # coerced internally
    "model" # coerced internally
  )

  paramtest = run_paramtest(learner, fun_list, exclude, tag = "train")
  expect_paramtest(paramtest)
})

test_that("paramtest surv.penalized predict", {
  learner = lrn("surv.penalized")
  fun_list = list(penalized:::predict) # nolint
  exclude = c(
    "object", # passed internally
    "unpenalized" # see learner details
  )

  paramtest = run_paramtest(learner, fun_list, exclude, tag = "predict")
  expect_paramtest(paramtest)
})