tests/testthat/test_base_resample_weights.R

In guillermozbta/mir: Machine Learning in R

context("resample_weights")

test_that("resample works with weights", {
  task = makeClassifTask(data = iris, target = "Species", weights = as.integer(iris$Species))
  res = resample(task = task, learner = "classif.rpart", resampling = makeResampleDesc("CV", iters = 2))
  expect_is(res$pred, "ResamplePrediction")
})