fix tests

Author: mattansb

R/standardize_info.R

@@ -320,7 +320,13 @@ standardize_info.default <- function(model,
 #' @keywords internal
 .std_info_response_smart <- function(model, info, data, model_matrix, types, robust = FALSE, w = NULL, ...) {
   if (info$is_linear) {
-    response <- insight::get_response(model)
+    if (inherits(model, c("gls", "lme"))) {
+      response <- insight::get_response(model)
+    } else if (inherits(model, "fixef")) {
+      response <- model.matrix(model, type = "lhs")
+    } else {
+      response <- stats::model.frame(model)[[1]]
+    }
 
     means <- deviations <- rep(NA_real_, length = length(names(model_matrix)))
     for (i in seq_along(names(model_matrix))) {
@@ -361,7 +367,13 @@ standardize_info.default <- function(model,
 
 #' @keywords internal
 .std_info_response_basic <- function(model, info, params, robust = FALSE, w = NULL, ...) {
-  response <- insight::get_response(model)
+  if (inherits(model, c("gls", "lme"))) {
+    response <- insight::get_response(model)
+  } else if (inherits(model, "fixef")) {
+    response <- model.matrix(model, type = "lhs")
+  } else {
+    response <- stats::model.frame(model)[[1]]
+  }
 
   if (info$is_linear) {
     if (robust) {

tests/testthat/test-standardize_parameters.R

@@ -178,23 +178,22 @@ test_that("standardize_parameters (with functions /  interactions)", {
   m1 <- lm(Y ~ X * Z)
   m2 <- lm(Y ~ X * scale(Z))
   m3 <- lm(Y ~ scale(X) * Z)
-  m4 <- lm(Y ~ scale(X) * scale(Z))
+  m4 <- lm(scale(Y) ~ scale(X) + scale(Z) + scale(scale(X) * scale(Z))) # ground truth
 
   expect_equal(
     standardize_parameters(m1, method = "basic")$Std_Coefficient,
+    model_parameters(m4)$Coefficient,
+    ignore_attr = TRUE
+  )
+  expect_equal(
     standardize_parameters(m2, method = "basic")$Std_Coefficient,
+    model_parameters(m4)$Coefficient,
     ignore_attr = TRUE
   )
   expect_equal(
-    standardize_parameters(m1, method = "basic")$Std_Coefficient,
     standardize_parameters(m3, method = "basic")$Std_Coefficient,
-    ignore_attr = TRUE
+    model_parameters(m4)$Coefficient
   )
-  # expect_equal(
-  #   standardize_parameters(m1, method = "basic")$Std_Coefficient,
-  #   standardize_parameters(m4, method = "basic")$Std_Coefficient
-  # )
-
 
   # transformed resp or pred should not affect
   mtcars$cyl_exp <- exp(mtcars$cyl)