tests/testthat/test-loadData.R

In jc-castillo/vaccineEarlyInvest: Investment in Early Manufacturing Capacity for Vaccines

#Test files for function loadData in models.R
test_that('loadData report error if the includeVaccines contains vaccines that are not prescribed',{
  par = Parameters$new()
  expect_error(loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'),includeVaccines = 1))
  expect_error(loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'),includeVaccines = c('BCG',123)))
  expect_error(loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'),includeVaccines = 'asd'))
})


test_that('includeVaccines works',{
  par = Parameters$new()
  d0 = loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'))
  d1 = loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'),includeVaccines = c('BCG'))
  expect_true('Live attenuated bacteria' %in% d1$Subcategory)
  expect_equal(setdiff(d1$Subcategory,d0$Subcategory),'Live attenuated bacteria')
})


test_that('loadData returns a data.table',{
  par = Parameters$new()
  d0 = loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'))
  d1 = loadData(par = par, candidateFile = system.file('extdata/vaccinesSummaryAug20.csv', package = 'vaccineEarlyInvest'),includeVaccines = c('BCG'))
  expect_equal(class(d0),class(d1))
  expect_equal(class(d0),c('data.table','data.frame'))
})

test_that('loadData works with no candidateFile entered',{
  par = Parameters$new()
  d0 = loadData(par = par)
  d1 = loadData(par = par,includeVaccines = c('BCG'))
  expect_equal(class(d0),c('data.table','data.frame'))
  expect_equal(class(d1),c('data.table','data.frame'))
})