inst/paramtest/test_paramtest_classif_randomForest.R

library(mlr3learners.randomforest)

test_that("classif.randomforest", {
  learner = lrn("classif.randomForest")
  fun = randomForest::randomForest
  exclude = c(
    "x" # handled via mlr3
  )

  ParamTest = run_paramtest(learner, fun, exclude)
  expect_true(ParamTest, info = paste0(
    "\nMissing parameters:\n",
    paste0("- '", ParamTest$missing, "'", collapse = "\n")))
})

test_that("predict classif.randomForest", {
  learner = lrn("classif.randomForest")
  fun = randomForest:::predict.randomForest
  exclude = c(
    "object", # handled via mlr3
    "newdata", # handled via mlr3
    "type" # handled via mlr3
  )

  ParamTest = run_paramtest(learner, fun, exclude)
  expect_true(ParamTest, info = paste0(
    "\nMissing parameters:\n",
    paste0("- '", ParamTest$missing, "'", collapse = "\n")))
})
mlr3learners/mlr3learners.randomForest documentation built on June 4, 2020, 9:21 a.m.