tests/testthat/test_transform_categorical.R

In ModelOriented/rSAFE: Surrogate-Assisted Feature Extraction

context("Testing safely_transform_categorical.R file")

source("objects_for_tests.R")


test_that("safely_transform_categorical", {
  expect_is(safely_transform_categorical(explainer_rf, "district"), "list")
  expect_is(safely_transform_categorical(explainer_rf_modified, "district"), "list")
  expect_error(safely_transform_categorical(1:10, "district"))
  expect_error(safely_transform_categorical(explainer_rf, "strange_variable"), "Wrong variable name!")
})