tests/testthat/test-load_forecasts_local_zoltar.R

In reichlab/covidHubUtils: Utility functions for the COVID-19 forecast hub

context("load_forecasts_local_zoltar")

library(covidHubUtils)
library(dplyr)
library(lubridate)

# Need to change these to valid paths
# path to local clone of zolpy repository.
local_zoltpy_path <- NA
# an absolute path to
# covidHubUtils/tests/testthat/test-data/test-load_forecasts_local_zoltar/db.sqlite3
zoltar_sqlite_file <- NA

skip_if_no_zoltpy_or_sqlite <- function() {
  if (!is.na(local_zoltpy_path) & !is.na(zoltar_sqlite_file)) {
    if (!dir.exists(local_zoltpy_path) |
      !file.exists(zoltar_sqlite_file)) {
      skip("zoltpy or sqlite data not available")
    }
  } else {
    skip("zoltpy or sqlite data not available")
  }
}

test_that("load_forecast from local zoltar works with window size", {
  skip_if_no_zoltpy_or_sqlite()
  all_forecasts <- covidHubUtils::load_forecasts(
    models = c("COVIDhub-ensemble", "COVIDhub-baseline"),
    dates = c("2021-07-27", "2021-08-03"),
    date_window_size = 1,
    locations = "US",
    source = "local_zoltar",
    local_zoltpy_path = local_zoltpy_path,
    zoltar_sqlite_file = zoltar_sqlite_file
  )

  expect_identical(
    all_forecasts %>%
      dplyr::distinct(model, forecast_date) %>%
      dplyr::arrange(model, forecast_date),
    tidyr::expand_grid(
      model = c("COVIDhub-baseline", "COVIDhub-ensemble"),
      forecast_date = lubridate::ymd(c("2021-07-26", "2021-08-02"))
    )
  )

  expect_true(unique(all_forecasts$location) == "US")
})

test_that("load_forecast from local zoltar works with one dates not available", {
  skip_if_no_zoltpy_or_sqlite()
  all_forecasts <- covidHubUtils::load_forecasts(
    models = c("COVIDhub-ensemble", "COVIDhub-baseline"),
    dates = c("2021-07-27", "2021-08-04"),
    date_window_size = 1,
    locations = "US",
    source = "local_zoltar",
    local_zoltpy_path = local_zoltpy_path,
    zoltar_sqlite_file = zoltar_sqlite_file
  )

  expect_identical(
    all_forecasts %>%
      dplyr::distinct(model, forecast_date) %>%
      dplyr::arrange(model, forecast_date),
    tidyr::expand_grid(
      model = c("COVIDhub-baseline", "COVIDhub-ensemble"),
      forecast_date = lubridate::ymd(c("2021-07-26"))
    )
  )

  expect_true(unique(all_forecasts$location) == "US")
})

test_that("load_forecast from local zoltar works without specified dates", {
  skip_if_no_zoltpy_or_sqlite()
  all_forecasts <- covidHubUtils::load_forecasts(
    models = c("COVIDhub-ensemble", "COVIDhub-baseline"),
    locations = "US",
    source = "local_zoltar",
    local_zoltpy_path = local_zoltpy_path,
    zoltar_sqlite_file = zoltar_sqlite_file
  )
  
  expect_identical(
    all_forecasts %>%
      dplyr::distinct(model, forecast_date) %>%
      dplyr::arrange(model, forecast_date),
    tidyr::expand_grid(
      model = c("COVIDhub-baseline", "COVIDhub-ensemble"),
      forecast_date = lubridate::ymd(c("2021-07-26", "2021-08-02"))
    )
  )
  
  expect_true(unique(all_forecasts$location) == "US")
})

test_that("load_forecast from local zoltar works without specified parameters", {
  skip_if_no_zoltpy_or_sqlite()
  all_forecasts <- covidHubUtils::load_forecasts(
    source = "local_zoltar",
    local_zoltpy_path = local_zoltpy_path,
    zoltar_sqlite_file = zoltar_sqlite_file
  )
  
  expect_identical(
    all_forecasts %>%
      dplyr::distinct(model, forecast_date) %>%
      dplyr::arrange(model, forecast_date),
    tidyr::expand_grid(
      model = c("COVIDhub-baseline", "COVIDhub-ensemble"),
      forecast_date = lubridate::ymd(c("2021-07-26", "2021-08-02"))
    )
  )
})