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 test_oxcgrt(self):
data_loader = DataLoader("input")
oxcgrt_data = data_loader.oxcgrt()
assert isinstance(oxcgrt_data, OxCGRTData)
assert isinstance(oxcgrt_data.citation, str)
df = oxcgrt_data.cleaned()
assert isinstance(df, pd.DataFrame)
assert set(df.columns) == set(OxCGRTData.OXCGRT_COLS)
subset_cols_set = set(OxCGRTData.OXCGRT_COLS_WITHOUT_COUNTRY)
subset_df = oxcgrt_data.subset(country="Japan")
assert set(subset_df.columns) == subset_cols_set
subset_df_iso = oxcgrt_data.subset(iso3="JPN")
assert set(subset_df_iso.columns) == subset_cols_set
def oxcgrt_data():
data_loader = DataLoader("input")
return data_loader.oxcgrt()
def test_oxcgrt_local_file_unexpected(self):
data_loader = DataLoader("input")
local_path = Path("input") / "covid_jpn_total.csv"
with pytest.raises(Exception):
data_loader.oxcgrt(local_file=local_path)
def test_oxcgrt_local_file(self):
data_loader = DataLoader("input")
local_path = Path("input") / "OxCGRT_latest.csv"
data_loader.oxcgrt(local_file=local_path)
local_file = str(local_path)
data_loader.oxcgrt(local_file=local_file)