inst/paramtest/test_paramtest_surv.akritas.R

In mlr3learners/mlr3learners.proba: Learners for mlr3proba

library(mlr3learners.proba)

test_that("surv.akritas", {
  learner = lrn("surv.akritas")
  fun = akritas
  exclude = c(
    "formula", # handled via mlr3
    "data", # handled via mlr3
    "time_variable", # handled via mlr3
    "status_variable", # handled via mlr3
    "x", # handled via mlr3
    "y", # handled via mlr3
    "..." # not used
  )

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

# example for checking a predict function of a learner
test_that("surv.akritas_predict", {
  learner = lrn("surv.akritas")
  fun = predict
  exclude = c(
    "object", # handled via mlr3
    "newdata", # handled via mlr3
    "type", # handled via mlr3
    "distr6", # handled via mlr3
    "times", # all times returned
    "..." # not used
  )

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