tests/testthat/test_pipeop_probregrcompositor.R

In mlr3proba: Probabilistic Supervised Learning for 'mlr3'

test_that("PipeOpProbregrCompositor - basic properties", {
  expect_pipeop(PipeOpProbregrCompositor$new())
  expect_pipeop(PipeOpProbregrCompositor$new(param_vals = list()))
})

task = tgen("friedman1")$generate(10)

test_that("PipeOpProbregrCompositor", {
  gr = mlr3pipelines::ppl("probregrcompositor", lrn("regr.featureless", predict_type = "se"),
    graph_learner = TRUE)
  p = gr$train(task)$predict(task)
  expect_true(distr6::testDistribution(p$distr))
})

test_that("no se", {
  expect_error(mlr3pipelines::ppl("probregrcompositor", lrn("regr.rpart"),
    graph_learner = TRUE), "does not support")
})

test_that("probregr_compose", {
  gr = mlr3pipelines::ppl("probregrcompositor",
    learner = lrn("regr.featureless", predict_type = "se"),
    dist = "Logistic", graph_learner = TRUE)$train(task)$predict(task)
  expect_true(distr6::testDistribution(gr$distr))
})