Skip to content

Commit b5e7d7a

Browse files
committed
Restore test_impact_forecast.py
1 parent 84b344b commit b5e7d7a

File tree

1 file changed

+6
-21
lines changed

1 file changed

+6
-21
lines changed

climada/engine/test/test_impact_forecast.py

Lines changed: 6 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -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
5860
def 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-
7964
class TestImpactForecastInit:
8065

8166
def assert_impact_kwargs(self, impact: Impact, **kwargs):

0 commit comments

Comments
 (0)