tests/testthat/test_mlr_learners_surv_coxph.R

In mlr3proba: Probabilistic Supervised Learning for 'mlr3'

test_that("autotest", {
  set.seed(1)
  learner = lrn("surv.coxph")
  expect_learner(learner)
  # no idea why weights check here fails
  result = suppressWarnings(run_autotest(learner, exclude = "weights", check_replicable = FALSE,
    N = 10))
  expect_true(result, info = result$error)
})

test_that("weights", {
  learner = lrn("surv.coxph")
  task = generate_tasks.LearnerSurv(learner)$weights
  suppressWarnings(learner$train(task))
  expect_equal(learner$model$weights, task$weights$weight)
})


test_that("missing", {
  set.seed(1)
  learner = lrn("surv.coxph")
  expect_error(learner$predict(tsk("lung")))
})