tests/testthat/test_surrogate.R

In compstat-lmu/surrogates: Construct surrogates from OpenML or local files

context("surrogate_class")

test_that("initialization", {
  ds = system.file("extdata", "glmnet_sample.csv", package = "surrogates")
  ps = get_param_set("glmnet")
  surr = Surrogate$new(oml_task_id = 3, base_learner = "regr.glmnet", eval_measure = "auc",
    param_set = ps, surrogate_learner = "regr.ranger",
    data_source = ds, load_fun = load_from_csv, save_path = tempdir())
  expect_r6(surr,"Surrogate")
})

test_that("print_fun", {
  ds = system.file("extdata/", "glmnet_sample.csv", package = "surrogates")
  ps = get_param_set("glmnet")
  surr = Surrogate$new(oml_task_id = 3, base_learner = "regr.glmnet", eval_measure = "auc",
    param_set = ps, surrogate_learner = "regr.ranger",
    data_source = ds, load_fun = load_from_csv, save_path = tempdir())
  expect_invisible(print(surr))
})

test_that("acquire_object_fun", {
  ds = system.file("extdata/", "glmnet_sample.csv", package = "surrogates")
  ps = get_param_set("glmnet")
  surr = Surrogate$new(oml_task_id = 3, base_learner = "regr.glmnet", eval_measure = "auc",
    param_set = ps, surrogate_learner = "regr.ranger",
    data_source = ds, load_fun = load_from_csv, save_path = tempdir())
  surr$acquire_rtask()
  expect_class(surr$rtask, "RegrTask")
  surr$acquire_model()
  expect_class(surr$model, "WrappedModel")
})