Skip to content

Commit 412fbd8

Browse files
committed
Add test for hazardForecast.concat
1 parent ef3551b commit 412fbd8

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

climada/hazard/test/test_forecast.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,3 +88,14 @@ def test_from_hazard(lead_time, member, hazard, haz_kwargs):
8888
npt.assert_array_equal(haz_fc_from_haz.lead_time, lead_time)
8989
npt.assert_array_equal(haz_fc_from_haz.member, member)
9090
assert_hazard_kwargs(haz_fc_from_haz, **haz_kwargs)
91+
92+
93+
def test_hazard_forecast_concat(haz_fc, lead_time, member):
94+
haz_fc1 = haz_fc.select(event_id=[1, 2])
95+
haz_fc2 = haz_fc.select(event_id=[3, 4])
96+
haz_fc_concat = HazardForecast.concat([haz_fc1, haz_fc2])
97+
assert isinstance(haz_fc_concat, HazardForecast)
98+
npt.assert_array_equal(
99+
haz_fc_concat.lead_time, np.concatenate([lead_time, lead_time])
100+
)
101+
npt.assert_array_equal(haz_fc_concat.member, np.concatenate([member, member]))

0 commit comments

Comments
 (0)