def test_filter_by_date_interval():
"""Test filter by date interval"""
df = filter_by_date_and_resolution(df_data, "2019-12-27/2019-12-29", Resolution.HOURLY)
assert not df.empty
df = filter_by_date_and_resolution(df_data, "2020/2022", Resolution.HOURLY)
assert df.empty
def test_filter_by_date():
"""Test filter by date"""
df = filter_by_date_and_resolution(df_data, "2019-12-28", Resolution.HOURLY)
assert not df.empty
df = filter_by_date_and_resolution(df_data, "2019-12-27", Resolution.HOURLY)
assert df.empty
def test_filter_by_date_annual():
df = pd.DataFrame.from_dict({
"station_id": ["01048"],
"dataset": ["climate_summary"],
"parameter": ["temperature_air_max_200"],
"from_date": [dateutil.parser.isoparse("2019-01-01T00:00:00.000Z")],
"to_date": [dateutil.parser.isoparse("2019-12-31T00:00:00.000Z")],
"value": [1.3],
"quality": [None],
})
df = filter_by_date_and_resolution(df,
date="2019-05/2019-09",
resolution=Resolution.ANNUAL)
assert not df.empty
df = filter_by_date_and_resolution(df,
date="2020/2022",
resolution=Resolution.ANNUAL)
assert df.empty
df = filter_by_date_and_resolution(df,
date="2020",
resolution=Resolution.ANNUAL)
assert df.empty
def test_filter_by_date_monthly():
result = pd.DataFrame.from_dict({
"station_id": ["01048"],
"dataset": ["climate_summary"],
"parameter": ["temperature_air_max_200"],
"from_date": [dateutil.parser.isoparse("2019-12-28T00:00:00.000Z")],
"to_date": [dateutil.parser.isoparse("2020-01-28T00:00:00.000Z")],
"value": [1.3],
"quality": [None],
})
df = filter_by_date_and_resolution(result, "2019-12/2020-01",
Resolution.MONTHLY)
assert not df.empty
df = filter_by_date_and_resolution(result, "2020/2022", Resolution.MONTHLY)
assert df.empty
df = filter_by_date_and_resolution(result, "2020", Resolution.MONTHLY)
assert df.empty
def filter_by_date(self, date: str) -> pd.DataFrame:
self.df = filter_by_date_and_resolution(
self.df, date=date, resolution=self.stations.resolution)
return self.df
