test_load_year.R
In compstatr: Tools for St. Louis Crime Data

context("test cs_load_year function")

## clean-up temp directories
tmpdir <- tempdir()
unlink(tmpdir, recursive = TRUE)

# load data and save items to temp directory ------------------------------------------------

## create temp directory with data subdir
tmpdir <- tempdir()
fs::dir_create(paste0(tmpdir,"/data/"))

## write files
cs_example(path = paste0(tmpdir, "/data/"))

## prep files
cs_prep_year(path = paste0(tmpdir, "/data/"))

# test function ------------------------------------------------

test_that("correctly specified functions execute without error", {
  expect_error(cs_load_year(path = paste0(tmpdir, "/data/")), NA)
})

# test output ------------------------------------------------

## create result
result <- cs_load_year(path = paste0(tmpdir,"/data"))

## create master and result for object names
names_master <- month.name
names_master <- names_master[order(names_master)]

names_result <- names(result)
names_result <- names_result[order(names_result)]

test_that("output is correct", {
  expect_equal(length(result), 12)
  expect_equal("cs_year_list" %in% class(result), TRUE)
  expect_equal("list" %in% class(result), TRUE)
  expect_equal(names_master, names_result)
})

# test parameters ------------------------------------------------

## add additional files
readr::write_csv(january2018, path = paste0(tmpdir,"/data/january2018.csv"))

test_that("too many files trigger error", {
  expect_error(cs_load_year(path = paste0(tmpdir,"/data")),
               "There are too many files in the specified folder. Load crime files in yearly batches of 12 monthly files.")
})

## delete files
fs::file_delete(paste0(tmpdir,"/data/december2017.csv"))
fs::file_delete(paste0(tmpdir,"/data/january2018.csv"))

test_that("too few files triggers warning", {
  expect_warning(cs_load_year(path = paste0(tmpdir,"/data")),
                 "There are fewer than 12 files in the specified folder. You are only loading a partial year.")
})

# final options ------------------------------------------------

unlink(tmpdir, recursive = TRUE)