revdep/checks.noindex/mlr3proba/old/mlr3proba.Rcheck/tests/testthat/test_mlr_learners_surv_rpart.R

In gbm-developers/gbm: Generalized Boosted Regression Models

context("surv.rpart")

test_that("autotest", {
  learner = mlr_learners$get("surv.rpart")
  expect_learner(learner)
  result = run_autotest(learner, check_replicable = FALSE)
  expect_true(result, info = result$error)
})


test_that("importance/selected", {
  learner = lrn("surv.rpart")
  expect_error(learner$importance(), "No model stored")
  expect_error(learner$selected_features(), "No model stored")
  learner$train(tsk("rats"))
  expect_silent(learner$selected_features())
  expect_silent(learner$importance())
})