tests/testthat/test-list.R

In tidypredict: Run Predictions Inside the Database

fitted_rf <- function(...) tidypredict_fit(parse_model(...))

skip_if_not_installed("randomForest")

test_that("Supports parsed models in list objects", {
  expect_equal(
    class(fitted_rf(lm(mpg ~ wt, data = mtcars)))[1],
    "call"
  )
  expect_equal(
    length(fitted_rf(
      randomForest::randomForest(Species ~ ., data = iris)
    )),
    500
  )
})