tests/testthat/test-clean_vars.R

In ggeffects: Create Tidy Data Frames of Marginal Effects for 'ggplot' from Model Outputs

skip_on_cran()
skip_on_os(c("mac", "solaris"))
skip_if_not_installed("emmeans")

test_that("ggpredict, clean_vars", {
  data(efc, package = "ggeffects")
  efc$c172code <- as.factor(efc$c172code)
  efc$e42dep <- as.factor(efc$e42dep)
  fit <- lm(barthtot ~ c12hour + c172code + e42dep, data = efc)
  expect_identical(nrow(ggpredict(fit, "c172code")), 3L)
  expect_identical(nrow(ggpredict(fit, "c172code [1,3]")), 2L)
  expect_identical(nrow(ggpredict(fit, "c172code[1,3]")), 2L)
  expect_identical(nrow(ggemmeans(fit, "c172code")), 3L)
  expect_identical(nrow(ggemmeans(fit, "c172code [1,3]")), 2L)
  expect_identical(nrow(ggemmeans(fit, "c172code[1,3]")), 2L)
  expect_identical(nrow(ggpredict(fit, "e42dep")), 4L)
  expect_identical(nrow(ggpredict(fit, "e42dep [1,3]")), 2L)
  expect_identical(nrow(ggpredict(fit, "e42dep[1,3]")), 2L)
  expect_identical(nrow(ggpredict(fit, c("c172code", "e42dep"))), 12L)
  expect_identical(nrow(ggpredict(fit, c("c172code [1,3]", "e42dep"))), 8L)
  expect_identical(nrow(ggpredict(fit, c("c172code", "e42dep [1,3]"))), 6L)
  expect_identical(nrow(ggpredict(fit, c("c172code [1,3]", "e42dep [1,3]"))), 4L)
  expect_identical(nrow(ggpredict(fit, c("c172code[1,3]", "e42dep"))), 8L)
  expect_identical(nrow(ggpredict(fit, c("c172code", "e42dep[1,3]"))), 6L)
  expect_identical(nrow(ggpredict(fit, c("c172code[1,3]", "e42dep[1,3]"))), 4L)
  expect_identical(nrow(ggemmeans(fit, c("c172code", "e42dep"))), 12L)
  expect_identical(nrow(ggemmeans(fit, c("c172code [1,3]", "e42dep"))), 8L)
  expect_identical(nrow(ggemmeans(fit, c("c172code", "e42dep [1,3]"))), 6L)
  expect_identical(nrow(ggemmeans(fit, c("c172code [1,3]", "e42dep [1,3]"))), 4L)
  expect_identical(nrow(ggemmeans(fit, c("c172code[1,3]", "e42dep"))), 8L)
  expect_identical(nrow(ggemmeans(fit, c("c172code", "e42dep[1,3]"))), 6L)
  expect_identical(nrow(ggemmeans(fit, c("c172code[1,3]", "e42dep[1,3]"))), 4L)
})