tests/testthat/test_bottom_code_quant.R

In Strabes/recipes.ext: Extensions to R's tidymodels

library(testthat)
library(recipes)
library(dplyr)
library(tidymodels.ext)

context("bottom_code_quant")

df <- tibble::tibble(
  x = seq(0,100,by=1),
  y = rep(c(1),101),
  z = seq(0,1,by=0.01)^2 + 3)

test_that("bottom coding by quantile",{
  rec_obj <- df |>
    recipe(y ~ .) |>
    step_bottom_code_quant(
      all_numeric_predictors(),prob = 0.1)

  rec_obj_prepped <- prep(rec_obj, training = df)

  res <- bake(rec_obj_prepped, df) |>
    select(x,y,z)

  answer <- df |>
    mutate(x = case_when(x < 10 ~ 10, T ~ x),
           z = case_when(z < 3.01 ~ 3.01, T ~z)) |>
    select(x,y,z)

  expect_equal(res,answer)
})

test_that("printing", {
  rec_obj <- df |>
    recipe(y ~ .)
  bottom_code <- rec_obj |>
    step_bottom_code_quant(
      all_numeric_predictors(),prob = 0.1)
  expect_output(print(bottom_code))
  expect_output(prep(bottom_code, training = df, verbose = TRUE))
})