tests/testthat/test_paramtest_RWeka_classif_IBk.R

In mlr-org/mlr3extralearners: Extra Learners For mlr3

test_that("paramtest classif.IBk train", {
  learner = lrn("classif.IBk")
  fun = RWeka::IBk
  exclude = weka_control_args(RWeka::IBk)
  # formula and data are handled via mlr3
  # mlr3 does not have the `control` argument because the parameters can be specified directly
  exclude = c("formula", "data", "control", "weight", exclude)
  paramtest = run_paramtest(learner, fun, exclude, tag = "train")
  expect_paramtest(paramtest)
  control_args = setdiff(weka_control_args(RWeka::IBk), c("I", "F"))
  expect_true(all(control_args %in% learner$param_set$ids()))
})

test_that("paramtest classif.IBk predict", {
  # Here we test that the learner implements those arguments that are passed via the
  # control argument to RWeka::IBk
  learner = lrn("classif.IBk")
  exclude = c( # all handled by mlr3
    "object",
    "newdata",
    "type"
  )
  paramtest = run_paramtest(learner, RWeka:::predict.Weka_classifier, exclude, tag = "predict") # nolint
  expect_paramtest(paramtest)
})