tests/testthat/test_preprocess.R

In mstaniak/egalitaRian: Compare Validation and Training Datasets and Explore Differences

context("Test preprocessing")

testthat::test_that("Additional columns in training data are detected", {
  testthat::expect_warning(check_data_validity(training_dataset,
                                               validation_dataset[, 2:21]))
  testthat::expect_is(egalitarian(training_dataset,
                                  validation_dataset[, 2:21], "y"),
                      "egalitarian")

})

testthat::test_that("Additional columns in validation data are detected", {
  expect_warning(check_data_validity(training_dataset[, 2:21],
                                     validation_dataset))
  testthat::expect_is(egalitarian(training_dataset[, 2:21],
                                  validation_dataset, "y"), "egalitarian")
})

testthat::test_that("Mismatched column classes are detected", {
  expect_warning(check_data_validity(mutate(training_dataset,
                                            fct = as.character(fct)),
                                     validation_dataset))
  testthat::expect_is(egalitarian(mutate(training_dataset,
                                         fct = as.character(fct)),
                                  validation_dataset, "y"), "egalitarian")
})

testthat::test_that("Unequal factor levels are detected", {
  expect_warning(check_data_validity(training_dataset,
                                     mutate(validation_dataset,
                                            fct = as.factor(as.character(
                                              rbinom(50, size = 7,
                                                     prob = 0.8))))))
})

testthat::test_that("Egalitarian object is created", {
  testthat::expect_is(egalitarian(training_dataset,
                                  validation_dataset, "y"), "egalitarian")
})

testthat::test_that("Generic functions work fine", {
  testthat::expect_output(print(egalitarian(training_dataset,
                                  validation_dataset, "y")))
  testthat::expect_output(print(egalitarian(training_dataset,
                                            validation_dataset, "y", regr_glm)))
})