tests/testthat/test-format_model.R

In report: Automated Reporting of Results and Statistical Models

test_that("format_model", {
  expect_identical(format_model(lm(Sepal.Length ~ Petal.Length * Species, data = iris)), "linear model")
  expect_identical(format_model(glm(vs ~ disp, data = mtcars, family = "binomial")), "logistic model")
  expect_identical(format_model(glm(vs ~ disp, data = mtcars, family = binomial(link = "probit"))), "probit model")
  expect_identical(format_model(glm(vs ~ mpg, data = mtcars, family = "poisson")), "poisson model")
})


test_that("format_model", {
  skip_if_not_installed("lme4")
  expect_identical(format_model(lme4::lmer(wt ~ cyl + (1 | gear), data = mtcars)), "linear mixed model")
  expect_identical(
    format_model(lme4::glmer(vs ~ cyl + (1 | gear), data = mtcars, family = "binomial")),
    "logistic mixed model"
  )
  expect_identical(
    format_model(lme4::glmer(vs ~ drat + cyl + (1 | gear), data = mtcars, family = "binomial")),
    "logistic mixed model"
  )
})




test_that("format_model", {
  skip_if_not_installed("rstanarm")
  expect_identical(
    format_model(
      suppressWarnings(
        rstanarm::stan_glm(mpg ~ wt, data = mtcars, refresh = 0, iter = 50)
      )
    ),
    "Bayesian linear model"
  )
  expect_identical(
    format_model(
      suppressWarnings(
        rstanarm::stan_glm(vs ~ wt,
          data = mtcars,
          family = "binomial",
          refresh = 0,
          iter = 50
        )
      )
    ),
    "Bayesian logistic model"
  )
})