tests/testthat/test_ppl.R

In mlr3pipelines: Preprocessing Operators and Pipelines for 'mlr3'

context("mlr_graphs")

test_that("mlr_graphs access works", {

  expect_equal(
    ppl("robustify"),
    mlr_graphs$get("robustify")
  )


  expect_equal(
    ppl("robustify", char_to_fct.affect_columns = selector_all()),
    {
      graph = mlr_graphs$get("robustify")
      graph$param_set$values$char_to_fct.affect_columns = selector_all()
      graph
    }
  )

  expect_equal(
    ppl("bagging", graph = po("nop")),
    mlr_graphs$get("bagging", graph = po("nop"))
  )

  expect_equal(ppl(), mlr_graphs)
})


test_that("mlr_pipeops multi-access works", {

  expect_equal(
    ppls("robustify"),
    list(mlr_graphs$get("robustify"))
  )


  expect_equal(
    ppls("robustify", char_to_fct.affect_columns = selector_all()),
    {
      graph = mlr_graphs$get("robustify")
      graph$param_set$values$char_to_fct.affect_columns = selector_all()
      list(graph)
    }
  )

  expect_equal(
    ppls("bagging", graph = po("nop")),
    list(mlr_graphs$get("bagging", graph = po("nop")))
  )

  expect_equal(
    ppls(c(x = "robustify")),
    list(x = mlr_graphs$get("robustify"))
  )


  expect_equal(ppls(), mlr_graphs)

})

test_that("mlr3book authors don't sleepwalk through life", {
  skip_if_not_installed("rpart")

  tasks = tsks(c("breast_cancer", "sonar"))

  glrn_stack = as_learner(ppl("robustify") %>>% ppl("stacking",
      lrns(c("classif.rpart", "classif.debug")),
      lrn("classif.rpart", id = "classif.rpart2")
  ))
  glrn_stack$id = "Stack"

  learners = c(glrn_stack)
  bmr = benchmark(benchmark_grid(tasks, learners, rsmp("cv", folds = 2)))

})