tests/testthat/test_automlr_mbo.R

In mlr-org/automlr: Automatic Machine Learning in R

context("mbo")

###
# Test 'easy' nofailSearchSpace

test_that("backend 'mbo' works with basic search space", {
  source("helper_automlrhelpers.R")
  backendToTest = "mbo"
  searchSpaceToTest = nofailSearchSpace
  checkBackend(searchSpaceToTest, backendToTest)
})

###
# Test withFailSearchSpace

test_that("backend 'mbo' works with search space with model failures allowed", {
  source("helper_automlrhelpers.R")
  backendToTest = "mbo"
  searchSpaceToTest = withFailSearchSpace
  checkBackend(searchSpaceToTest, backendToTest, learnersMayFail = TRUE)
})

###
# Test withPPSearchSpace, including a wrapper

test_that("backend 'mbo' works with search space with preprocessing and model failure", {
  source("helper_automlrhelpers.R")
  backendToTest = "mbo"
  searchSpaceToTest = withPPSearchSpace
  checkBackend(searchSpaceToTest, backendToTest, thorough = TRUE, learnersMayFail = TRUE)
})

###
# Test paramtestSearchSpace, with parameters of different types

test_that("backend 'mbo' works with search space with various parameter types", {
  source("helper_automlrhelpers.R")
  backendToTest = "mbo"
  searchSpaceToTest = paramtestSearchSpace
  checkBackend(searchSpaceToTest, backendToTest)
})

###
# Test reqstestSearchSpace, with complicated parameter requirements

test_that("backend 'mbo' works with search space with requirements", {
  source("helper_automlrhelpers.R")
  backendToTest = "mbo"
  searchSpaceToTest = reqstestSearchSpace
  checkBackend(searchSpaceToTest, backendToTest)
})