library(tibble)
library(dplyr)
n <- 20
ex_dat <-
data.frame(x1 = seq(0, 1, length = n), x2 = rep(1:5, 4)) %>%
group_by(x1)
# Test fixes for Issue #125
test_that("grouped data frames work", {
n <- 20
rec <- recipe(~., data = ex_dat) %>%
step_poly(x2, degree = 2)
rec_trained <- prep(rec, training = ex_dat, verbose = FALSE)
rec_trans <- bake(rec_trained, new_data = ex_dat)
expect_equal(names(rec_trans), c("x1", "x2_poly_1", "x2_poly_2"))
})
Generated by dwww version 1.15 on Sat May 18 03:45:52 CEST 2024.