mmrm() models

DESCRIPTION

@@ -61,6 +61,7 @@ Suggests:
     MASS,
     mgcv,
     mice,
+    mmrm (>= 0.3.6),
     multgee,
     nnet,
     ordinal,

NEWS.md

@@ -1,5 +1,9 @@
 # broom.helpers (development version)
 
+**New supported models**
+
+- Support for `mmrm::mmrm()` models (#228)
+
 **New features**
 
 - new `tidy_post_fun` argument in `tidy_plus_plus()` (#235)

R/model_get_weights.R

@@ -4,7 +4,7 @@
 #'
 #' @param model a model object
 #' @note
-#' For class `svrepglm` objects (glm on a survey object with replicate weights),
+#' For class `svrepglm` objects (GLM on a survey object with replicate weights),
 #' it will return the original sampling weights of the data, not the replicate
 #' weights.
 #' @export

data-raw/DATASET.R

@@ -50,6 +50,7 @@ supported_models <-
     "`logitr::logitr()`", "Requires logitr >= 0.8.0",
     "`multgee::nomLORgee()`", "Experimental support. Use `tidy_multgee()` as `tidy_fun`.",
     "`multgee::ordLORgee()`", "Experimental support. Use `tidy_multgee()` as `tidy_fun`.",
+    "`mmrm::mmrm()`", "",
     "`pscl::zeroinfl()`", "Use `tidy_zeroinfl()` as `tidy_fun`.",
     "`pscl::hurdle()`", "Use `tidy_zeroinfl()` as `tidy_fun`.",
     "`betareg::betareg()`", "Use `tidy_parameters()` as `tidy_fun` with `component` argument to control with coefficients to return. `broom::tidy()` does not support the `exponentiate` argument for betareg models, use `tidy_parameters()` instead." # nolint

tests/testthat/test-tidy_plus_plus.R

@@ -911,6 +911,38 @@ test_that("tidy_plus_plus() works with betareg::betareg() models", {
   expect_equal(nrow(res), 24)
 })
 
+test_that("tidy_plus_plus() works with mmrm::mmrm() models", {
+  skip_on_cran()
+  skip_if_not_installed("mmrm")
+
+  m1 <- mmrm::mmrm(FEV1 ~ SEX + ARMCD + AVISIT + us(AVISIT | USUBJID), data = mmrm::fev_data)
+  m2 <- mmrm::mmrm(FEV1 ~ SEX + ARMCD * AVISIT + us(AVISIT | USUBJID), data = mmrm::fev_data)
+
+  expect_error(
+    res <- m1 %>% tidy_plus_plus(intercept = TRUE),
+    NA
+  )
+  expect_equal(nrow(res), 9)
+
+  expect_error(
+    res <- m1 %>% tidy_plus_plus(add_header_rows = TRUE),
+    NA
+  )
+  expect_equal(nrow(res), 11)
+
+  expect_error(
+    res <- m2 %>% tidy_plus_plus(intercept = TRUE),
+    NA
+  )
+  expect_equal(nrow(res), 12)
+
+  expect_error(
+    res <- m2 %>% tidy_plus_plus(add_header_rows = TRUE),
+    NA
+  )
+  expect_equal(nrow(res), 15)
+})
+
 test_that("tidy_post_fun argument of `tidy_plus_plus()`", {
   mod <- lm(Petal.Length ~ Petal.Width + Species, iris)