fix tests after insight-update

easystats · Feb 11, 2025 · b04ea19 · b04ea19
1 parent 475813a
commit b04ea19
Show file tree

Hide file tree

Showing 7 changed files with 93 additions and 28 deletions.
diff --git a/tests/testthat/_snaps/plot/plot-cat-num-predictor-3.svg b/tests/testthat/_snaps/plot/plot-cat-num-predictor-3.svg
diff --git a/tests/testthat/_snaps/plot/plot-cat-num-predictor-4.svg b/tests/testthat/_snaps/plot/plot-cat-num-predictor-4.svg
diff --git a/tests/testthat/_snaps/plot/plot-cat-num-predictor-5.svg b/tests/testthat/_snaps/plot/plot-cat-num-predictor-5.svg
diff --git a/tests/testthat/test-estimate_expectation.R b/tests/testthat/test-estimate_expectation.R
@@ -46,7 +46,7 @@ test_that("estimate_expectation - data-grid", {
 
   m <- lm(mpg ~ cyl, data = mtcars)
   estim <- estimate_expectation(m, data = "grid")
-  expect_identical(dim(estim), c(10L, 5L))
+  expect_identical(dim(estim), c(3L, 5L))
   expect_named(
     estim,
     c("cyl", "Predicted", "SE", "CI_low", "CI_high")

diff --git a/tests/testthat/test-estimate_predicted.R b/tests/testthat/test-estimate_predicted.R
@@ -146,13 +146,13 @@ test_that("estimate_expectation - Frequentist", {
 
   model <- lme4::lmer(wt ~ cyl + (1 | gear), data = data)
   estim <- estimate_link(model)
-  expect_equal(dim(estim), c(10, 6))
+  expect_identical(dim(estim), c(3L, 6L))
   estim <- estimate_expectation(model)
   expect_equal(dim(estim), c(32, 7))
 
   model <- lme4::glmer(vs ~ cyl + (1 | gear), data = data, family = "binomial")
   estim <- estimate_link(model)
-  expect_equal(dim(estim), c(10, 6))
+  expect_identical(dim(estim), c(3L, 6L))
   estim <- estimate_expectation(model)
   expect_equal(dim(estim), c(32, 7))
 })

diff --git a/tests/testthat/test-multivariate_response.R b/tests/testthat/test-multivariate_response.R
@@ -20,5 +20,5 @@ test_that("multivariate response", {
   expect_identical(dim(out), c(64L, 4L))
 
   out <- estimate_expectation(fm1, by = "cyl")
-  expect_identical(dim(out), c(20L, 4L))
+  expect_identical(dim(out), c(6L, 4L))
 })
diff --git a/tests/testthat/test-plot.R b/tests/testthat/test-plot.R
@@ -287,6 +287,12 @@ test_that("plots, numeric or categorical predictors are detected", {
     "plot-cat-num-predictor-4",
     plot(pr, show_data = FALSE)
   )
+  pr <- estimate_expectation(m_cat, by = "cyl", range = "grid")
+  set.seed(123)
+  vdiffr::expect_doppelganger(
+    "plot-cat-num-predictor-5",
+    plot(pr, show_data = FALSE)
+  )
 })