skip_on_os(c("mac", "solaris"))
skip_if_not_installed("ggplot2")
skip_if_not_installed("datawizard")
skip_if_not_installed("vdiffr")

test_that("plot, vignette", {
  data(efc, package = "ggeffects")
  efc$c172code <- datawizard::to_factor(efc$c172code)
  fit <- lm(barthtot ~ c12hour + neg_c_7 + c161sex + c172code, data = efc)

  dat <- predict_response(fit, terms = c("c12hour", "c172code"))
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-1",
    plot(dat, facets = TRUE, show_data = TRUE, verbose = FALSE)
  )

  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-1-shape",
    plot(dat, facets = TRUE, show_data = TRUE, dot_shape = 21, verbose = FALSE)
  )

  # don't use facets, b/w figure, w/o confidence bands
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-2",
    plot(dat, colors = "bw", show_ci = FALSE, show_data = TRUE, verbose = FALSE)
  )

  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-2-shape",
    plot(dat, colors = "bw", show_ci = FALSE, show_data = TRUE, dot_shape = 2, verbose = FALSE)
  )

  set.seed(123)
  dat <- predict_response(fit, terms = c("c12hour", "c172code"))
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-3",
    plot(dat, show_data = TRUE, verbose = FALSE)
  )

  # for three variables, automatic facetting
  ## FIXME: doesn't work
  # dat <- predict_response(fit, terms = c("c12hour", "c172code", "c161sex"))
  # set.seed(123)
  # vdiffr::expect_doppelganger(
  #   "Vignette-plotintro-4",
  #   plot(dat, show_data = TRUE)
  # )

  dat <- predict_response(fit, terms = c("c172code", "c161sex"))
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-5",
    plot(dat, show_data = TRUE, jitter = TRUE, verbose = FALSE)
  )

  # point-geoms for discrete x-axis can be connected with lines
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-6",
    plot(dat, connect_lines = TRUE, show_data = TRUE, jitter = TRUE, verbose = FALSE)
  )

  ## FIXME: doesn't work
  # # for four variables, automatic facetting and integrated panel
  # dat <- predict_response(fit, terms = c("c12hour", "c172code", "c161sex", "neg_c_7"))
  # # use 'one_plot = FALSE' for returning multiple single plots
  # set.seed(123)
  # vdiffr::expect_doppelganger(
  #   "Vignette-plotintro-7",
  #   plot(dat, one_plot = TRUE, show_data = TRUE)
  # )

  # dashed lines for CI
  dat <- predict_response(fit, terms = "c12hour")
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-9",
    plot(dat, ci_style = "dash", show_data = TRUE, verbose = FALSE)
  )

  # facet by group
  dat <- predict_response(fit, terms = c("c12hour", "c172code"))
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-10",
    plot(dat, facets = TRUE, ci_style = "errorbar", dot_size = 1.5, show_data = TRUE, verbose = FALSE) # nolint
  )

  dat <- predict_response(fit, terms = "c172code")
  set.seed(123)
  vdiffr::expect_doppelganger(
    "Vignette-plotintro-11",
    plot(dat, facets = TRUE, ci_style = "errorbar", dot_size = 1.5, show_data = TRUE, jitter = TRUE, verbose = FALSE) # nolint
  )
})