Fix get_data()

easystats · Apr 26, 2024 · 197ccfd · 197ccfd
1 parent 63d8646
commit 197ccfd
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 1 deletion.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,7 +1,7 @@
 Type: Package
 Package: insight
 Title: Easy Access to Model Information for Various Model Objects
-Version: 0.19.10.1
+Version: 0.19.10.2
 Authors@R: 
     c(person(given = "Daniel",
              family = "Lüdecke",

diff --git a/NAMESPACE b/NAMESPACE
@@ -388,6 +388,7 @@ S3method(get_data,clm2)
 S3method(get_data,clmm)
 S3method(get_data,clmm2)
 S3method(get_data,complmrob)
+S3method(get_data,coxme)
 S3method(get_data,coxph)
 S3method(get_data,cpglmm)
 S3method(get_data,data.frame)

diff --git a/NEWS.md b/NEWS.md
@@ -5,6 +5,11 @@
 * More informative error message for `get_varcov()` when the requested
   `vcov`-function failed.
 
+## Bug fixes
+
+* Fixed issue with `get_data()` for `coxme` models when `source`was set to
+  `"modelframe"`.
+
 # insight 0.19.10
 
 ## Bug fixes

diff --git a/R/get_data.R b/R/get_data.R
@@ -1844,6 +1844,9 @@ get_data.coxph <- function(x, source = "environment", verbose = TRUE, ...) {
   dat
 }
 
+#' @export
+get_data.coxme <- get_data.coxph
+
 
 #' @export
 get_data.BGGM <- function(x, ...) {

diff --git a/tests/testthat/test-coxme.R b/tests/testthat/test-coxme.R
@@ -183,5 +183,21 @@ withr::with_environment(
       expect_identical(find_statistic(m1), "z-statistic")
       expect_identical(find_statistic(m2), "z-statistic")
     })
+
+    # get_data() works with this example
+    test_that("get_data works with mf", {
+      data(eortc, package = "coxme")
+      d <- eortc
+      d$surv <- survival::Surv(d$y, d$uncens)
+      mcoxme <- coxme::coxme(surv ~ trt + (1 | center), data = d)
+      # environment
+      out <- get_data(mcoxme)
+      expect_identical(nrow(out), 2323L)
+      expect_named(out, c("surv", "trt", "center"))
+      # modelframe
+      out <- get_data(mcoxme, source = "mf")
+      expect_identical(nrow(out), 2323L)
+      expect_named(out, c("y", "uncens", "center", "trt", "surv"))
+    })
   }
 )