Hazard forecast test select (#1175)

* Add HazardForecast.select test

---------

Co-authored-by: luseverin <[email protected]>
Co-authored-by: Lukas Riedel <[email protected]>

Author: luseverin

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