tests/testthat/test-extract_model_data.R

In m-clark/mixedup: Miscellaneous functions for mixed models

context('test extract_model_data')


# Test initial checks -----------------------------------------------------


test_that('extract_model_data errors with wrong model', {
  mod = lm(mpg ~ vs, mtcars)
  expect_error(extract_model_data(mod))
})


# standard models ---------------------------------------------------------

# these have a model.frame method

test_that('extract_model_data works', {
  expect_s3_class(extract_model_data(lmer_1), 'data.frame')
})
test_that('extract_model_data works', {
  expect_s3_class(extract_model_data(brm_1), 'data.frame')
})
test_that('extract_model_data works', {
  expect_s3_class(extract_model_data(stan_glmer_1), 'data.frame')
})
test_that('extract_model_data works', {
  expect_s3_class(extract_model_data(tmb_1), 'data.frame')
})
test_that('extract_model_data works', {
  expect_s3_class(extract_model_data(gam_1), 'data.frame')
})

# Non-standard models -----------------------------------------------------


test_that('extract_model_data works with lme', {
  expect_s3_class(extract_model_data(lme_1), 'data.frame')
})

# these don't
test_that('extract_model_data works with lme', {
  expect_error(extract_model_data(nlme_1))
})

# stanreg multivariate
test_that('extract_model_data works with stanregmv', {
  expect_equal(length(extract_model_data(stan_glmer_mv)), 2)
})

test_that('extract_model_data works with stanregmv', {
  expect_equal(length(extract_model_data(stan_glmer_jm)), 3)
})