tests/testthat/test-get_titles.R

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

if (suppressWarnings(
  require("testthat") &&
  require("ggeffects") &&
  require("haven") &&
  require("sjlabelled") &&
  require("sjmisc")
)) {
  data(efc)

  efc$c172code <- sjmisc::to_factor(efc$c172code)
  fit <- lm(barthtot ~ c12hour + neg_c_7 + c161sex + c172code, data = efc)
  mydf <- ggpredict(fit, terms = c("c12hour", "c161sex", "c172code"))

  test_that("ggpredict, get_x_title", {
    expect_equal(get_x_title(mydf), "average number of hours of care per week")
  })

  test_that("ggpredict, get_y_title", {
    expect_equal(get_y_title(mydf), "Total score BARTHEL INDEX")
  })

  test_that("ggpredict, get_legend_labels", {
    expect_equal(get_legend_labels(mydf), c("Male", "Female"))
  })

  test_that("ggpredict, get_legend_title", {
    expect_equal(get_legend_title(mydf), "carer's gender")
  })

  mydf <- ggpredict(fit, terms = "c172code")

  test_that("ggpredict, get_x_labels", {
    expect_equal(get_x_labels(mydf), c("low level of education", "intermediate level of education", "high level of education"))
  })
}