tests/testthat/test-remove_intercept.R

test_that("tidy_remove_intercept() works for basic models", {
  mod <- glm(response ~ stage + grade + trt, gtsummary::trial, family = binomial)
  res <- mod %>%
    tidy_and_attach() %>%
    tidy_remove_intercept()
  expect_equal(
    res %>% dplyr::filter(var_type == "intercept") %>% nrow(),
    0L
  )
})


test_that("test tidy_remove_intercept() checks", {
  mod <- glm(response ~ stage + grade + trt, gtsummary::trial, family = binomial)
  # expect an error if no model attached
  expect_error(mod %>% broom::tidy() %>% tidy_remove_intercept())

  # could be apply twice (no error)
  expect_error(
    mod %>% tidy_and_attach() %>% tidy_remove_intercept() %>% tidy_remove_intercept(),
    NA
  )
})

Try the broom.helpers package in your browser

Any scripts or data that you put into this service are public.

broom.helpers documentation built on Aug. 7, 2023, 5:08 p.m.