tests/testthat/test-data-grid.R

In modelr: Modelling Functions that Work with the Pipe

test_that("can generate typical values", {
  df <- tibble::tibble(
    x = rep(c("a", "b"), each = 5),
    y = rep(c("a", "b"), 5),
    z = rnorm(10)
  )
  mod <- lm(z ~ x + y, data = df)
  out <- data_grid(df, .model = mod)

  expect_equal(out$x, c("a", "a", "b", "b"))
  expect_equal(out$y, c("a", "b", "a", "b"))
})

test_that("data_grid() returns a tibble", {
  mod <- lm(mpg ~ wt, data = mtcars)
  out <- data_grid(mtcars, .model = mod)
  expect_s3_class(out, "tbl_df")
})