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")