fix, add test

Author: strengejacke

R/estimate_grouplevel.R

@@ -75,7 +75,7 @@ estimate_grouplevel <- function(model, type = "random", dispersion = TRUE, test
   )
 
   # get cleaned parameter names with additional information
-  clean_parameters <- attributes(params)$clean_parameters$Group
+  clean_parameters <- attributes(params)$clean_parameters
 
   ## TODO: for now, rstanarm has no random effects on the sigma parameter
   ## however, if this changes, we need another solution here (and in

tests/testthat/test-estimate_grouplevel.R

@@ -1,6 +1,11 @@
 skip_if(utils::packageVersion("insight") <= "1.1.0")
 skip_if(utils::packageVersion("parameters") <= "0.24.1")
 
+test_that("estimate_grouplevel - errors", {
+  m <- lm(mpg ~ gear, data = mtcars)
+  expect_error(estimate_grouplevel(m), regex = "Model must be a mixed")
+})
+
 test_that("estimate_grouplevel - lme4", {
   skip_if_not_installed("lme4")
   set.seed(333)
@@ -83,7 +88,7 @@ test_that("estimate_grouplevel - glmmTMB", {
   expect_named(out, c("Group", "Level", "Parameter", "Coefficient"))
 })
 
-test_that("estimate_grouplevel - Bayesian", {
+test_that("estimate_grouplevel - Bayesian, brms", {
   skip_on_cran()
   skip_if_not_installed("curl")
   skip_if_offline()
@@ -112,3 +117,33 @@ test_that("estimate_grouplevel - Bayesian", {
   expect_identical(dim(out), c(12L, 9L))
   expect_named(out, c("Component", "Group", "Level", "Parameter", "Median", "MAD", "CI", "CI_low", "CI_high"))
 })
+
+test_that("estimate_grouplevel - Bayesian, rstanarm", {
+  skip_on_cran()
+  skip_if_not_installed("curl")
+  skip_if_offline()
+  skip_if_not_installed("httr2")
+  skip_if_not_installed("rstanarm")
+
+  m <- insight::download_model("brms_mixed_10")
+  skip_if(is.null(m))
+
+  out <- estimate_grouplevel(m)
+  expect_identical(dim(out), c(6L, 8L))
+  expect_named(out, c("Group", "Level", "Parameter", "Median", "MAD", "CI", "CI_low", "CI_high"))
+
+  out <- estimate_grouplevel(m, type = "total", dispersion = FALSE)
+  expect_identical(dim(out), c(6L, 7L))
+  expect_named(out, c("Group", "Level", "Parameter", "Median", "CI", "CI_low", "CI_high"))
+
+  m <- insight::download_model("brms_sigma_3")
+  skip_if(is.null(m))
+
+  out <- estimate_grouplevel(m)
+  expect_identical(dim(out), c(12L, 9L))
+  expect_named(out, c("Component", "Group", "Level", "Parameter", "Median", "MAD", "CI", "CI_low", "CI_high"))
+
+  out <- estimate_grouplevel(m, type = "total")
+  expect_identical(dim(out), c(12L, 9L))
+  expect_named(out, c("Component", "Group", "Level", "Parameter", "Median", "MAD", "CI", "CI_low", "CI_high"))
+})