tests/testthat/test_mbo.R

In jakob-r/mboBench: Benchmark functions for MBO

context("mbo benchmark")

test_that("mbo benchmark works", {
  skip_if_not_installed("mlrMBO")
  library(mlrMBO)
  benchmark = generateSimpleBenchmark(makeBraninFunction())
  ctrl = makeMBOControl()
  ctrl = setMBOControlTermination(ctrl, more.termination.conds = benchmark$mlrmbo.termination.criterions)
  lrn = makeLearner("regr.glm", predict.type = "se", config = list(show.learner.output = FALSE))
  bench.res = lapply(1:2, function(i) {
    run = mbo(
      fun = benchmark$smoof.fun,
      design = benchmark$getInitialDesignEvaluated(i),
      control = ctrl,
      learner = lrn)
    expect_equal(run$final.state, "term.feval")
    expect_equal(getOptPathLength(run$opt.path), benchmark$termination.criterions$evals$vars$max.evals)
    BenchResult$new(
      id = NULL,
      algo.name = "mlrMBO",
      benchmark = benchmark,
      opt.path = run$opt.path,
      repl = i
    )
  })
  brv = BenchReplVis$new(bench.res, benchmark)
  brv$plot_opt_path_progress()
  brv$plot_threshold_progress()

})