tests/testthat/test-plot_estimates.R

In EpiNow2: Estimate Real-Time Case Counts and Time-Varying Epidemiological Parameters

fit <- readRDS("test-models/estimate_infections/static.rds")
cases <- EpiNow2::example_confirmed[1:30]

test_that("plot_estimates can plot by type", {
  expect_error(
    plot_estimates(
      estimate = fit$summarised[variable == "infections"],
      ylab = "Cases", max_plot = 2
    ),
    NA
  )
  expect_error(
    plot_estimates(
      estimate = fit$summarised[variable == "infections"],
      reported = cases,
      ylab = "Cases", max_plot = 2
    ),
    NA
  )
  expect_error(
    plot_estimates(
      estimate = fit$summarised[variable == "infections"],
      reported = cases,
      ylab = "Cases", max_plot = 2
    ) +
      ggplot2::facet_wrap(~type, scales = "free_y"),
    NA
  )
})

test_that("plot_estimates can add a horizontal line ", {
  expect_error(
    plot_estimates(
      estimate = fit$summarised[variable == "R"],
      ylab = "Effective Reproduction No.",
      hline = 1
    ),
    NA
  )
})

test_that("plot_estimates can be restricted to only plot within the forecasting
           time horizon", {
  expect_error(
    plot_estimates(
      estimate = fit$summarised[variable == "R"],
      ylab = "Effective Reproduction No.",
      estimate_type = "Estimate"
    ),
    NA
  )
})