tests/testthat/test_benchmark.R

In mlr-org/mlrng: Machine Learning in R - Next Generation

context("benchmark")

test_that("benchmark", {
  tasks = lapply(c("iris", "sonar"), mlr.tasks$get)
  lrn1 = mlr.learners$get("classif.rpart")
  lrn1$par.vals = list(cp = 0.01)
  lrn2 = mlr.learners$get("classif.dummy")
  learners = list(lrn1, lrn2)
  resamplings = list(mlr.resamplings$get("cv"))
  resamplings[[1]]$iters = 3L
  measures = list(mlr.measures$get("mmce"))

  withr::with_options(list(mlrng.train.encapsulation = "none"), {
    bmr = benchmark(tasks, learners, resamplings, measures)
  })
  expect_result(bmr)
  expect_data_table(bmr$data, nrow = 12)
})