@@ -56,7 +56,6 @@ def impact_forecast(impact, lead_time, member):
5656
5757
5858class TestImpactForecastInit :
59-
6059 def assert_impact_kwargs (self , impact : Impact , ** kwargs ):
6160 for key , value in kwargs .items ():
6261 attr = getattr (impact , key )
@@ -87,7 +86,6 @@ def test_impact_forecast_from_impact(
8786
8887def test_impact_forecast_select (impact_forecast , lead_time , member , impact_kwargs ):
8988 """Check if Impact.select works on the derived class"""
90-
9189 event_ids = impact_kwargs ["event_id" ][np .array ([2 , 0 ])]
9290 impact_fc = impact_forecast .select (event_ids = event_ids )
9391 # NOTE: Events keep their original order
@@ -96,3 +94,12 @@ def test_impact_forecast_select(impact_forecast, lead_time, member, impact_kwarg
9694 )
9795 npt .assert_array_equal (impact_fc .member , member [np .array ([0 , 2 ])])
9896 npt .assert_array_equal (impact_fc .lead_time , lead_time [np .array ([0 , 2 ])])
97+
98+
99+ @pytest .mark .skip ("Concat from base class does not work" )
100+ def test_impact_forecast_concat (impact_forecast , member ):
101+ """Check if Impact.concat works on the derived class"""
102+ impact_fc = ImpactForecast .concat (
103+ [impact_forecast , impact_forecast ], reset_event_ids = True
104+ )
105+ npt .assert_array_equal (impact_fc .member , np .concatenate ([member , member ]))
0 commit comments