Skip to content

Commit d04fecf

Browse files
luseverinpeanutfun
andauthored
Hazard forecast concat test (#1178)
* Add test for hazardForecast.concat * Skip HazardForecast.concat test --------- Co-authored-by: luseverin <[email protected]> Co-authored-by: Lukas Riedel <[email protected]>
1 parent 4aaad4e commit d04fecf

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

climada/hazard/test/test_forecast.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,18 @@ def test_from_hazard(lead_time, member, hazard, haz_kwargs):
8888
assert_hazard_kwargs(haz_fc_from_haz, **haz_kwargs)
8989

9090

91+
@pytest.mark.skip("Concat from base class does not work")
92+
def test_hazard_forecast_concat(haz_fc, lead_time, member):
93+
haz_fc1 = haz_fc.select(event_id=[1, 2])
94+
haz_fc2 = haz_fc.select(event_id=[3, 4])
95+
haz_fc_concat = HazardForecast.concat([haz_fc1, haz_fc2])
96+
assert isinstance(haz_fc_concat, HazardForecast)
97+
npt.assert_array_equal(
98+
haz_fc_concat.lead_time, np.concatenate([lead_time, lead_time])
99+
)
100+
npt.assert_array_equal(haz_fc_concat.member, np.concatenate([member, member]))
101+
102+
91103
def test_hazard_forecast_select(haz_fc, lead_time, member):
92104
"""Check if Hazard.select works on the derived class"""
93105
haz_fc_select = haz_fc.select(event_id=[4, 1])

0 commit comments

Comments
 (0)