Merge branch 'impact-forecast-select' into impact-forecast-concat

Author: peanutfun

climada/engine/test/test_impact_forecast.py

@@ -42,7 +42,7 @@ def impact(impact_kwargs):
 
 @pytest.fixture
 def lead_time():
-    return pd.date_range("2000-01-01", "2000-01-02", periods=6).to_numpy()
+    return pd.timedelta_range(start="1 day", periods=6).to_numpy()
 
 
 @pytest.fixture

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