tests/testthat/test-character_covariates.R

In jeremyrcoyle/sl3: Pipelines for Machine Learning and Super Learning

context("test-character_covariates.R -- Automatically cast character covariates to factors")

test_that("character covariates are cast to factors", {
  data(cpp_imputed)
  covars <- c("apgar1", "apgar5", "parity", "gagebrth", "mage", "meducyrs", "sexn")
  outcome <- "haz"
  cpp_imputed$sexn <- as.character(cpp_imputed$sexn)

  expect_warning(
    task_character_to_factor <- make_sl3_Task(cpp_imputed, covars, outcome),
    "Character variables found: sexn;\nConverting these to factors"
  )

  expect_equal(class(task_character_to_factor$get_node("covariates")$sexn), "factor")
})