tests/testthat/test-smooth-coefs.R

In gavinsimpson/gratia: Graceful 'ggplot'-Based Graphics and Other Functions for GAMs Fitted Using 'mgcv'

# Test smooth_coefs() and related functions

test_that("smooth_coefs() works with a gam", {
  expect_silent(b <- smooth_coefs(m_gam, select = "s(x0)"))
  expect_named(b, paste("s(x0)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth_coefs() throws error without select", {
  expect_error(
    smooth_coefs(m_gam),
    "argument \"select\" is missing, with no default"
  )
})

test_that("smooth_coefs() works with a gamm", {
  expect_silent(b <- smooth_coefs(m_gamm, select = "s(x0)"))
  expect_named(b, paste("s(x0)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth_coefs() works with a gamm4", {
  expect_silent(b <- smooth_coefs(m_gamm4, select = "s(x0)"))
  expect_named(b, paste("s(x0)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth_coefs() works with a proper gamm4 class", {
  expect_silent(b <- smooth_coefs(m_gamm4_real, select = "s(x0)"))
  expect_named(b, paste("s(x0)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth_coefs() works with a bam", {
  expect_silent(b <- smooth_coefs(m_bam, select = "s(x0)"))
  expect_named(b, paste("s(x0)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth_coefs() works with a scam", {
  expect_silent(b <- smooth_coefs(m_scam, select = "s(x2)"))
  expect_named(b, paste("s(x2)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth_coefs() works with a mgcv.smooth", {
  sm <- get_smooth(m_gam, term = "s(x0)")
  expect_silent(b <- smooth_coefs(sm, model = m_gam))
  expect_named(b, paste("s(x0)", 1:9, sep = "."))
  expect_identical(length(b), 9L)
})

test_that("smooth-coefs print() output is as expected", {
  skip_on_ci()
  expect_snapshot({
    print(smooth_coefs(m_gam, select = "s(x0)"))
  })
})

test_that("term argument deprecated in smooth-coefs", {
  expect_warning(smooth_coefs(m_gam, term = "s(x0)"), "deprecated")
  expect_warning(smooth_coefs(m_gamm, term = "s(x0)"), "deprecated")
  expect_warning(smooth_coefs(m_gamm4, term = "s(x0)"), "deprecated")
  expect_warning(smooth_coefs(m_scam, term = "s(x1)"), "deprecated")
})