def test_dwd_observation_stations_nearby_number_multiple(): request = DwdObservationRequest( DwdObservationDataset.TEMPERATURE_AIR, DwdObservationResolution.HOURLY, DwdObservationPeriod.HISTORICAL, datetime(2020, 1, 1), datetime(2020, 1, 20), ) nearby_station = request.filter_by_rank( 50.0, 8.9, 3, ) nearby_station = nearby_station.df.drop("to_date", axis="columns") assert_frame_equal(nearby_station, EXPECTED_STATIONS_DF)
def test_dwd_observation_stations_nearby_number_single(): # Test for one nearest station request = DwdObservationRequest( DwdObservationDataset.TEMPERATURE_AIR, DwdObservationResolution.HOURLY, DwdObservationPeriod.RECENT, datetime(2020, 1, 1), datetime(2020, 1, 20), ) nearby_station = request.filter_by_rank( 50.0, 8.9, 1, ) nearby_station = nearby_station.df.drop("to_date", axis="columns") assert_frame_equal(nearby_station, EXPECTED_DF_SINGLE)