tests/testthat/test_sugar.R

In mlr3tuningspaces: Search Spaces for 'mlr3'

test_that("lts dispatch works", {
  # missing dispatch
  dict = lts()
  expect_class(dict, "DictionaryTuningSpaces")

  # character dispatch
  tuning_space = lts("classif.rpart.default")
  expect_r6(tuning_space, "TuningSpace")

  # learner dispatch
  learner = lts(lrn("classif.rpart"))
  expect_learner(learner)
})

test_that("ltss sugar function works", {
  # character dispatch
  tuning_spaces = ltss(c("classif.rpart.default", "classif.svm.default"))
  expect_list(tuning_spaces, len = 2)
  expect_r6(tuning_spaces[[1]], "TuningSpace")
  expect_r6(tuning_spaces[[2]], "TuningSpace")

  # learner dispatch
  learners = ltss(list(lrn("classif.rpart"), lrn("classif.svm")))
  expect_learner(learners[[1]])
  expect_learner(learners[[2]])
})

test_that("constants can be set", {
  learner = lts(lrn("classif.rpart", xval = 1))
  expect_learner(learner)
  expect_equal(learner$param_set$values$xval, 1)
  expect_class(learner$param_set$values$minsplit, "RangeTuneToken")

  learners = ltss(list(lrn("classif.rpart", xval = 1), lrn("classif.svm", type = "C-classification")))
  expect_learner(learners[[1]])
  expect_learner(learners[[2]])
  expect_equal(learners[[1]]$param_set$values$xval, 1)
  expect_class(learners[[1]]$param_set$values$minsplit, "RangeTuneToken")
  expect_equal(learners[[2]]$param_set$values$type, "C-classification")
  expect_class(learners[[2]]$param_set$values$degree, "RangeTuneToken")
})

test_that("modifying tuning space works", {
  # add
  tuning_space = lts("classif.rpart.default", maxdepth = to_tune())
  expect_r6(tuning_space, "TuningSpace")
  expect_names(names(tuning_space$values), permutation.of = c("minsplit", "minbucket", "cp", "maxdepth"))
  expect_list(tuning_space$values, types = "TuneToken")

  # remove
  tuning_space = lts("classif.rpart.default", minsplit = NULL)
  expect_r6(tuning_space, "TuningSpace")
  expect_names(names(tuning_space$values), permutation.of = c("minbucket", "cp"))
  expect_list(tuning_space$values, types = "TuneToken")

  # overwrite
  tuning_space = lts("classif.rpart.default", minsplit = to_tune(64, 128))
  expect_r6(tuning_space, "TuningSpace")
  expect_names(names(tuning_space$values), permutation.of = c("minsplit", "minbucket", "cp"))
  expect_list(tuning_space$values, types = "TuneToken")
  expect_equal(tuning_space$values$minsplit$content$lower, 64)
})