tests/testthat/test-onehot_encoder.R

In UBC-MDS/encodeR: A collection of categorical encoders in R

test_that("onehot_encoder() does not produce a list with correct dimensions", {

  testing_data <- readr::read_csv("../testdata/testing_data.csv")

  train <- testing_data %>%
    dplyr::filter(train_test_1 == 'train')

  test <- testing_data %>%
    dplyr::filter(train_test_1 == 'test')

  result_test <- onehot_encoder(train, test, cat_columns = c("feature_cat_chr"))
  result <- onehot_encoder(train, cat_columns = c("feature_cat_chr"))

  expect_equal(length(result_test),2)

  expect_equal(length(result),1)
})

test_that("onehot_encoder() does not throw an error if we input invalid
          arguments", {

  testing_data <- readr::read_csv("../testdata/testing_data.csv")

  # Test for categorical columns not being in X_train
  expect_error(onehot_encoder(
    X_train = testing_data,
    X_test = testing_data,
    cat_columns = c("cool4cats")))

  # check type of input X_train
  expect_error(onehot_encoder(
    X_train = c(1,2,3),
    X_test = testing_data,
    cat_columns = c("feature_cat_chr")))
})