tests/testthat/test-plot_wis.R

In epiforecasts/scoringutils: Utilities for Scoring and Assessing Predictions

sum_scores <- suppressMessages(
  summarise_scores(scores_quantile, by = c("model", "target_type"))
)

test_that("plot_wis() throws an error if WIS components are missing", {
  ex_score <- na.omit(example_quantile) %>%
    as_forecast() %>%
    score(metrics = metrics_quantile(select = "wis"))

  expect_error(
    plot_wis(ex_score),
    "Columns 'overprediction', 'underprediction', 'dispersion' not found in data."
  )
})


test_that("plot_wis() works as expected with relative contributions", {
  p <- plot_wis(sum_scores,
    x = "model",
    relative_contributions = TRUE
  ) +
    facet_wrap(~target_type)
  expect_s3_class(p, "ggplot")
  skip_on_cran()
  vdiffr::expect_doppelganger("plot_wis", p)
})

test_that("plot_wis() works as expected without relative contributions", {
  p <- plot_wis(sum_scores,
    x = "model",
    relative_contributions = FALSE
  ) +
    facet_wrap(~target_type)
  expect_s3_class(p, "ggplot")
  skip_on_cran()
  vdiffr::expect_doppelganger("plot_wis_no_relative", p)
})

test_that("plot_wis() works as expected when flipped", {
  p <- plot_wis(sum_scores,
    x = "model",
    relative_contributions = TRUE,
    flip = TRUE
  ) +
    facet_wrap(~target_type)
  expect_s3_class(p, "ggplot")
  skip_on_cran()
  vdiffr::expect_doppelganger("plot_wis_flip", p)
})