tests/testthat/test_surv_cvcoxboost.R

In mlr3learners/mlr3learners.coxboost: Learners from the {CoxBoost} package for 'mlr3'

context("surv.cvcoxboost")

test_that("autotest", {
  set.seed(1)
  learner = lrn("surv.cvcoxboost", maxstepno = 1, K = 2)
  expect_learner(learner)
  result = run_autotest(learner, N = 10,check_replicable = FALSE)
  expect_true(result, info = result$error)
})

test_that("optim", {
  set.seed(1)
  learner = lrn("surv.cvcoxboost",
    penalty = "optimCoxBoostPenalty",
    maxstepno = 1, minstepno = 0, iter.max = 1, K = 2)
  task = tgen("simsurv")$generate(10)
  expect_prediction_surv(suppressWarnings(learner$train(task)$predict(task)))
})