diff --git a/climada/hazard/test/test_forecast.py b/climada/hazard/test/test_forecast.py index 640818b396..6c8b29c2bb 100644 --- a/climada/hazard/test/test_forecast.py +++ b/climada/hazard/test/test_forecast.py @@ -86,3 +86,12 @@ def test_from_hazard(lead_time, member, hazard, haz_kwargs): npt.assert_array_equal(haz_fc_from_haz.lead_time, lead_time) npt.assert_array_equal(haz_fc_from_haz.member, member) assert_hazard_kwargs(haz_fc_from_haz, **haz_kwargs) + + +def test_hazard_forecast_select(haz_fc, lead_time, member): + """Check if Hazard.select works on the derived class""" + haz_fc_select = haz_fc.select(event_id=[4, 1]) + # NOTE: Events keep their original order + npt.assert_array_equal(haz_fc_select.event_id, haz_fc.event_id[np.array([3, 0])]) + npt.assert_array_equal(haz_fc_select.member, member[np.array([3, 0])]) + npt.assert_array_equal(haz_fc_select.lead_time, lead_time[np.array([3, 0])])