tests/testthat/test_read_data_jhu_csse.R

In jimtyhurst/covid19: COVID-19 Exploratory Data Analysis

library(dplyr)
library(lubridate)
library(testthat)

context("Reading COVID-19 case data from JHU CSSE")

test_that("Expected columns are present for confirmed cases", {
  expected_columns <- c(
    "country_region",
    "province_state",
    "lat",
    "long",
    "date",
    "cumulative_total"
  )
  actual_columns <- colnames(read_confirmed_cases_jhu_csse())
  expect_gte(length(actual_columns), length(expected_columns))
  expect_true(all(expected_columns %in% actual_columns))
})

test_that("confirmed cases is not empty", {
  expect_gt(dim(read_confirmed_cases_jhu_csse())[1], 0)
})

test_that("US has expected 2 total cases on 2020-01-24", {
  expected_total <- 2
  actual_total <- read_confirmed_cases_jhu_csse() %>%
    dplyr::filter(country_region == "US", date == lubridate::ymd("2020-01-24")) %>%
    dplyr::select(cumulative_total) %>%
    getElement("cumulative_total")
  expect_equal(actual_total, expected_total)
})