tests/testthat/test_predictor.R

In mcboost: Multi-Calibration Boosting

skip_if_not_installed("mlr3")

test_that("Predictor class instantiation", {
  skip_on_cran()
  prd = Predictor$new()
  expect_class(prd, "Predictor")
  expect_error(prd$fit(), fixed = "Abstract base class")
  expect_error(prd$predict(), fixed = "Abstract base class")
})

test_that("ConstantPredictor", {
  skip_on_cran()
  prd = ConstantPredictor$new(0.7)
  expect_class(prd, "ConstantPredictor")
  expect_true(prd$is_fitted)
  p = prd$predict(data.frame(x = 1:3))
  expect_equal(p, rep(0.7, 3))
})

test_that("LearnerPredictor - response", {
  skip_on_cran()
  skip_if_not_installed("mlr3learners") 
  prd = LearnerPredictor$new(lrn("classif.rpart"))
  expect_class(prd, "LearnerPredictor")
  prd$fit(iris[1:100,1:4], factor(iris$Species[1:100]))
  expect_true(prd$is_fitted)
  p = prd$predict(iris[,1:4])
  expect_numeric(p, len = 150L, lower = 0, upper = 1)
})

test_that("LearnerPredictor - probs", {
  skip_on_cran()
  skip_if_not_installed("mlr3learners") 
  prd = LearnerPredictor$new(lrn("classif.rpart", predict_type = "prob"))
  expect_class(prd, "LearnerPredictor")
  prd$fit(iris[51:150,1:4], factor(iris$Species[51:150]))
  expect_true(prd$is_fitted)
  p = prd$predict(iris[,1:4])
  expect_numeric(p, len = 150L, lower = 0, upper = 1)
})