tests/testthat/test_gam_tidiers.R

In exploratory-io/exploratory_func: R package for Exploratory

# how to run this test:
# devtools::test(filter="gam_tidiers")

context("test tidiers for gam")

test_that("test glance, augment", {
  # TODO: with columns with space, error happens.
  #mtcars2 <- mtcars %>% rename(`cy l`=cyl, `mp g`=mpg)
  #gam_model <- mgcv::gam(data=mtcars2, `cy l`~`mp g`)
  gam_model <- mgcv::gam(data=mtcars, cyl~mpg)
  class(gam_model) <- c("gam_exploratory",class(gam_model))
  res <- broom::glance(gam_model)
  expect_equal(class(res), "data.frame")
  expect_equal(class(res$r_squared), "numeric")
  expect_equal(class(res$deviance_explained), "numeric")
  expect_equal(class(res$gcv), "numeric")
  expect_equal(class(res$scale_est), "numeric")

  # Test augment too.
  res <- broom::augment(gam_model, newdata=mtcars, se=TRUE)
  expect_equal(class(res$.fitted[[1]]), "numeric")
  expect_equal(class(res$.se.fit[[1]]), "numeric")
})