inst/paramtest/test_paramtest_surv_blackboost.R

library(mlr3learners.mboost)

test_that("surv.blackboost", {
  learner = lrn("surv.blackboost")
  fun = mboost::blackboost
  exclude = c(
    "formula", # handled via mlr3
    "data", # handled via mlr3
    "weights", # handled via mlr3
    "control", # handled to mboost::boost_control
    "tree_controls", # handled internally
    "..." # not used
  )

  ParamTest = run_paramtest(learner, fun, exclude)
  expect_true(ParamTest, info = paste0(
    "\nMissing parameters:\n",
    paste0("- '", ParamTest$missing, "'", collapse = "\n")))
})

test_that("surv.blackboost_boost_control", {
  learner = lrn("surv.blackboost")
  fun = mboost::boost_control
  exclude = c(
    "center" # deprecated
  )

  ParamTest = run_paramtest(learner, fun, exclude)
  expect_true(ParamTest, info = paste0(
    "\nMissing parameters:\n",
    paste0("- '", ParamTest$missing, "'", collapse = "\n")))
})
mlr3learners/mlr3learners.mboost documentation built on Aug. 9, 2020, 10:47 p.m.