tests/testthat/test-name_to_fips.R

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

context("name_to_fips")
library(covidHubUtils)
library(dplyr)
library(testthat)
library(lubridate)
library(stringr)

test_that("name_to_fips works as with full names as inputs",{
  test_data_US <- c("Hampshire County, MA","Bullock County, AL","New Jersey","United States")
  expected_data_US <- c("US","34","01011","25015")
  function_output <- name_to_fips(test_data_US, hub = c("US"))
  expect_equal(function_output , expected_data_US)
  
  test_data_Flu <- c("New Jersey","United States")
  expected_data_Flu <- c("34","US")
  function_output <- name_to_fips(test_data_Flu, hub = c("FluSight"))
  expect_equal(function_output , expected_data_Flu)
  
  test_data_ECDC <- c("Belgium","Spain","Italy","United Kingdom")
  expected_data_ECDC <- c("BE","ES","IT","GB")
  function_output <- name_to_fips(test_data_ECDC, hub = c("ECDC"))
  expect_equal(function_output , expected_data_ECDC)
  
})

test_that("name_to_fips throws error with invalid input",{
  test_data_US <- c("Hampshre, MA","Bullock County","Nw Jeey","Uned States")
  expect_error(name_to_fips(test_data_US, hub = c("US")))
  test_data_Flu <- c("Nw Jeey","Uned States")
  expect_error(name_to_fips(test_data_Flu, hub = c("FluSight")))
  test_data_ECDC <- c("Begium","Sain","I-taly","Unted Kingdom")
  expect_error(name_to_fips(test_data_ECDC, hub = c("ECDC")))
}
)


test_that("NULL input should throw NULL",{
  test_data = NULL
  expect_null(name_to_fips(test_data, hub = "US"))
  expect_null(name_to_fips(test_data, hub = "ECDC"))
  expect_null(name_to_fips(test_data, hub = "FluSight"))
  
})

test_that("name_to_fips works as with full names and fips/CBSA code as inputs",{
  test_data_US <- c("Hampshire County, MA","01001","New Jersey","United States")
  expected_data_US <- c("US","34","25015","01001")
  function_output <- name_to_fips(test_data_US, hub = c("US"))
  expect_equal(function_output , expected_data_US)
  
  test_data_Flu <- c("34","United States")
  expected_data_Flu <- c("US","34")
  function_output <- name_to_fips(test_data_Flu, hub = c("FluSight"))
  expect_equal(function_output , expected_data_Flu)
  
  test_data_ECDC <- c("BE","Spain","IT","United Kingdom")
  expected_data_ECDC <- c("ES","GB","BE","IT")
  function_output <- name_to_fips(test_data_ECDC, hub = c("ECDC"))
  expect_equal(function_output , expected_data_ECDC)
})

test_that("name_to_fips works as with fips/CBSA code as inputs",{
  test_data_US <- c("01001","US")
  expected_data_US <- c("US","01001")
  function_output <- name_to_fips(test_data_US, hub = c("US"))
  expect_equal(function_output , expected_data_US)
  
  test_data_Flu <- c("34","US")
  expected_data_Flu <- c("US","34")
  function_output <- name_to_fips(test_data_Flu, hub = c("FluSight"))
  expect_equal(function_output , expected_data_Flu)
  
  test_data_ECDC <- c("BE","IT")
  expected_data_ECDC <- c("BE","IT")
  function_output <- name_to_fips(test_data_ECDC, hub = c("ECDC"))
  expect_equal(function_output , expected_data_ECDC)
})