def test_cleaning(self, jhu_data):
assert isinstance(jhu_data.raw, pd.DataFrame)
with pytest.raises(ValueError):
jhu_data.cleaned(population=None)
df = jhu_data.cleaned()
assert set(df.columns) == set(Term.COLUMNS)
assert isinstance(JHUData.from_dataframe(df), JHUData)
def test_from_dataframe(self, jhu_data):
df = jhu_data.cleaned()
jhu_data2 = JHUData.from_dataframe(df)
assert set(df.columns) == set(jhu_data2.cleaned().columns)
def test_from_dataframe(self, japan_data):
df = japan_data.cleaned()
jhu_data_df = JHUData.from_dataframe(df, directory="input_dir")
assert isinstance(jhu_data_df, JHUData)
assert jhu_data_df.directory == "input_dir"
jhu_data_df.records("Japan")
def test_from_dataframe(self, japan_data):
df = japan_data.cleaned()
jhu_data_df = JHUData.from_dataframe(df)
assert isinstance(jhu_data_df, JHUData)
jhu_data_df.records("Japan")