@@ -45,37 +45,22 @@ def impact(impact_kwargs):
4545
4646
4747@pytest .fixture
48- def lead_time ():
49- return pd .timedelta_range (start = "1 day" , periods = 6 ).to_numpy ()
48+ def lead_time (impact_kwargs ):
49+ return pd .timedelta_range (
50+ start = "1 day" , periods = len (impact_kwargs ["event_id" ])
51+ ).to_numpy ()
5052
5153
5254@pytest .fixture
53- def member ():
54- return np .arange (6 )
55+ def member (impact_kwargs ):
56+ return np .arange (len ( impact_kwargs [ "event_id" ]) )
5557
5658
5759@pytest .fixture
5860def impact_forecast (impact , lead_time , member ):
5961 return ImpactForecast .from_impact (impact , lead_time = lead_time , member = member )
6062
6163
62- class TestImpactForecastInit :
63-
64- @pytest .fixture
65- def lead_time (impact_kwargs ):
66- return pd .timedelta_range (
67- start = "1 day" , periods = len (impact_kwargs ["event_id" ])
68- ).to_numpy ()
69-
70- @pytest .fixture
71- def member (impact_kwargs ):
72- return np .arange (len (self , impact_kwargs ["event_id" ]))
73-
74- @pytest .fixture
75- def impact_forecast (impact , lead_time , member ):
76- return ImpactForecast .from_impact (impact , lead_time = lead_time , member = member )
77-
78-
7964class TestImpactForecastInit :
8065
8166 def assert_impact_kwargs (self , impact : Impact , ** kwargs ):
0 commit comments