inst/tinytest/test_step_intercept.R

In jkennel/hydrorecipes: Hydrogeology steps

formula <- as.formula(y~x)
rows <- 20

dat <- data.frame(x = rnorm(rows),
                  y = as.numeric(1:rows),
                  z = rnorm(rows))

#^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
frec = hydrorecipes:::Recipe$new(formula = formula, data = dat)$
  add_step(hydrorecipes:::StepIntercept$new())$
  plate("tbl")[,c(3,1,2)]
frec[[1]] <- as.integer(frec[[1]])

rec  = recipes::recipe(formula = formula, data = dat) |>
  recipes::step_intercept() |>
  recipes::prep() |>
  recipes::bake(new_data = NULL)

expect_equivalent(frec, rec, info = "hydrorecipes matches recipes")


#^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
frec2 = recipe(formula = formula, data = dat)|>
  step_intercept() |>
  plate("df")

frec1 = hydrorecipes:::Recipe$new(formula = formula, data = dat)$
  add_step(hydrorecipes:::StepIntercept$new())$
  plate("df")

expect_equivalent(frec1, frec2,
                            info = "R6 and hydrorecipes api are equivalent")