tests/testthat/test-extract-helpers.R

In tidymodels/tune: Tidy Tuning Tools

test_that("extract methods for last_fit objects", {
  lm_spec <- parsnip::linear_reg() %>% parsnip::set_engine("lm")
  lm_prn_fit <- parsnip::fit(lm_spec, mpg ~ ., data = mtcars)
  lm_wflow <-
    workflow() %>%
    add_model(lm_spec) %>%
    add_formula(mpg ~ .)
  lm_res <- last_fit(lm_wflow, split = rsample::initial_split(mtcars))

  expect_true(inherits(extract_fit_engine(lm_res), "lm"))
  expect_true(inherits(extract_fit_parsnip(lm_res), "model_fit"))
  expect_true(inherits(extract_mold(lm_res), "list"))
  expect_true(inherits(extract_preprocessor(lm_res), "formula"))
  expect_true(inherits(extract_spec_parsnip(lm_res), "model_spec"))
  expect_true(inherits(extract_workflow(lm_res), "workflow"))
})

test_that("extract methods for resample_results objects", {
  lm_spec <- parsnip::linear_reg() %>% parsnip::set_engine("lm")
  lm_rec_wflow <-
    workflow() %>%
    add_model(lm_spec) %>%
    add_recipe(recipes::recipe(mpg ~ ., data = mtcars) %>%
      recipes::step_normalize(recipes::all_numeric_predictors()))
  lm_rec_res <- fit_resamples(
    lm_rec_wflow,
    resamples = rsample::vfold_cv(mtcars, v = 2),
    control = control_resamples(save_workflow = TRUE)
  )

  expect_true(inherits(extract_recipe(lm_rec_res, estimated = FALSE), "recipe"))
  expect_true(!extract_recipe(lm_rec_res, estimated = FALSE)$steps[[1]]$trained)
  expect_true(inherits(extract_workflow(lm_rec_res), "workflow"))
})