tests/testthat/test-list.R

In edgararuiz/tidypredict: Run Predictions Inside the Database

context("test-list")

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

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