tests/testthat/test-target_encoding_lab.R

In collinear: Automated Multicollinearity Management

testthat::test_that("`target_encoding_lab()` works", {

  data(vi, vi_predictors)
  vi <- vi[1:1000, ]

  # Call the target_encoding_lab function
  encoded_df <- target_encoding_lab(
    df = vi,
    response = "vi_numeric",
    predictors = "koppen_zone",
    methods = c("mean", "rank", "loo"),
    smoothing = c(0, 30),
    white_noise = c(0, 0.1, 0.2),
    quiet = TRUE
  )

  # Check if the result is a data frame
  testthat::expect_true(
    is.data.frame(encoded_df)
    )

  # Check if the encoded variables have been added
  testthat::expect_true(
    "koppen_zone__encoded_mean" %in% colnames(encoded_df)
    )

  testthat::expect_true(
    "koppen_zone__encoded_rank" %in% colnames(encoded_df)
    )

  testthat::expect_true(
    "koppen_zone__encoded_loo" %in% colnames(encoded_df)
    )

  # Check if encoding methods have been applied
  testthat::expect_true(
    is.numeric(encoded_df$koppen_zone__encoded_mean)
    )

  testthat::expect_true(
    is.numeric(encoded_df$koppen_zone__encoded_rank)
    )

  testthat::expect_true(
    is.numeric(encoded_df$koppen_zone__encoded_loo)
    )

})