tests/testthat/test-verify_bins.R

In reichlab/cdcForecastUtils: Utility functions for CDC forecasts

context("verify_bins")

test_that("Correct entries are successful.",{
  expect_true(verify_bins(full_entry_new))
  expect_true(verify_bins(full_entry_state_new, challenge = "state_ili"))
  expect_true(verify_bins(hosp_template, challenge = "hospitalization"))
  
})


test_that("Missing bins report errors.", {
  rand_target <- sample(full_entry_new$target, 1)
  
  valid_bins <- unique(full_entry_new$bin[full_entry_new$target ==
                                                      rand_target & 
                                                       full_entry_new$type == 
                                                      "bin"])
  
  for (i in seq_along(valid_bins)) {
    tmp_entry <- full_entry_new[full_entry_new$bin != valid_bins[i], ]
    expect_warning(verify_bins(tmp_entry))
  }
})

test_that("Extra bin reports warning.", {
  rand_target <- sample(full_entry_new$target, 1)
  
  extra_row <- head(full_entry_new[full_entry_new$target == rand_target &
                                     full_entry_new$type == "bin", ], n = 1)
  extra_row$bin <- "extra"
  tmp_entry <- rbind(full_entry_new, extra_row)
  
  expect_error(verify_bins(tmp_entry))
})