tests/testthat/test-pull.R

In workflowsets: Create a Collection of 'tidymodels' Workflows

library(parsnip)
library(rsample)

lr_spec <- linear_reg() %>% set_engine("lm")

set.seed(1)
car_set_1 <-
  workflow_set(
    list(reg = mpg ~ ., nonlin = mpg ~ wt + 1 / sqrt(disp)),
    list(lm = lr_spec)
  ) %>%
  workflow_map("fit_resamples",
    resamples = vfold_cv(mtcars, v = 3),
    control = tune::control_resamples(save_pred = TRUE)
  )

# ------------------------------------------------------------------------------

test_that("pulling objects", {
  expect_warning(
    expect_equal(
      car_set_1 %>% pull_workflow("reg_lm"),
      car_set_1$info[[1]]$workflow[[1]]
    ),
    "deprecated"
  )

  expect_warning(
    expect_equal(
      car_set_1 %>% pull_workflow_set_result("reg_lm"),
      car_set_1$result[[1]]
    ),
    "deprecated"
  )

  expect_warning(
    expect_error(
      car_set_1 %>% pull_workflow_set_result("Gideon Nav"),
      "No workflow ID found"
    ),
    "deprecated"
  )

  expect_warning(
    expect_error(
      car_set_1 %>% pull_workflow("Coronabeth Tridentarius"),
      "No workflow ID found"
    ),
    "deprecated"
  )
})