def test_dataloader(self):
# Create DataLoader instance
with pytest.raises(TypeError):
DataLoader(directory=0)
data_loader = DataLoader(directory="input", update_interval=12)
# List of primary sources of COVID-19 Data Hub
assert data_loader.covid19dh_citation
# Data loading
assert isinstance(data_loader.jhu(), JHUData)
assert isinstance(data_loader.population(), PopulationData)
assert isinstance(data_loader.oxcgrt(), OxCGRTData)
assert isinstance(data_loader.japan(), CountryData)
# With local files
data_loader.jhu(local_file="input/covid19dh.csv")
data_loader.population(local_file="input/covid19dh.csv")
data_loader.oxcgrt(local_file="input/covid19dh.csv")
data_loader.japan(local_file="input/covid_jpn_total.csv")
def test_dataloader(self, jhu_data, population_data, oxcgrt_data,
japan_data, linelist_data, pcr_data, vaccine_data):
# List of primary sources of COVID-19 Data Hub
data_loader = DataLoader()
assert data_loader.covid19dh_citation
# Data loading
assert isinstance(jhu_data, JHUData)
assert isinstance(population_data, PopulationData)
assert isinstance(oxcgrt_data, OxCGRTData)
assert isinstance(japan_data, CountryData)
assert isinstance(linelist_data, LinelistData)
assert isinstance(pcr_data, PCRData)
assert isinstance(vaccine_data, VaccineData)
# Local file
data_loader.jhu(local_file="input/covid19dh.csv")
data_loader.population(local_file="input/covid19dh.csv")
data_loader.oxcgrt(local_file="input/covid19dh.csv")
data_loader.pcr(local_file="input/covid19dh.csv")
def population_data():
data_loader = DataLoader("input")
return data_loader.population()