tests/testthat/test_infill_opt_ea.R

In mlr-org/mlrMBO: Bayesian Optimization and Model-Based Optimization of Expensive Black-Box Functions

context("infillopt ea")

test_that("infillopt ea", {
  obj.fun = testf.fsphere.2d

  ctrl = makeMBOControl()
  ctrl = setMBOControlTermination(ctrl, iters = 4L)
  ctrl = setMBOControlInfill(ctrl, crit = crit.ei, opt = "ea", opt.restarts = 2L,
    opt.ea.maxit = 75L, opt.ea.lambda = 1L)

  des = testd.fsphere.2d
  res = mbo(obj.fun, des, control = ctrl)
  expect_true(res$y < 0.5, info = "real")

  obj.fun = smoof::makeSingleObjectiveFunction(
    fn = function(x) x$num1^2 + x$int1,
    par.set = makeParamSet(
      makeNumericParam("num1", lower = -1, upper = 1),
      makeIntegerParam("int1", lower = 0, upper = 2)
    ),
    has.simple.signature = FALSE
  )

  des = generateTestDesign(10L, getParamSet(obj.fun))
  res = mbo(obj.fun, des, control = ctrl)
  expect_true(res$y < 0.5, info = "integer mixed")
})