Skip to content

Commit 3b077a0

Browse files
committed
Add fixture for impact kwargs
1 parent b7b6545 commit 3b077a0

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

climada/engine/test/test_impact_forecast.py

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,17 @@
2727

2828
from climada.engine import Impact, ImpactForecast
2929

30-
from .test_impact import impact_kwargs
30+
from .test_impact import impact_kwargs as imp_kwargs
3131

3232

3333
@pytest.fixture
34-
def impact():
35-
return Impact(**impact_kwargs())
34+
def impact_kwargs():
35+
return imp_kwargs()
36+
37+
38+
@pytest.fixture
39+
def impact(impact_kwargs):
40+
return Impact(**impact_kwargs)
3641

3742

3843
def assert_impact_kwargs(impact: Impact, **kwargs):
@@ -50,20 +55,20 @@ class TestImpactForecastInit:
5055
lead_time = pd.date_range("2000-01-01", "2000-01-02", periods=6).to_numpy()
5156
member = np.arange(6)
5257

53-
def test_impact_forecast_init(self):
58+
def test_impact_forecast_init(self, impact_kwargs):
5459
forecast1 = ImpactForecast(
5560
lead_time=self.lead_time,
5661
member=self.member,
57-
**impact_kwargs(),
62+
**impact_kwargs,
5863
)
5964
npt.assert_array_equal(forecast1.lead_time, self.lead_time)
6065
npt.assert_array_equal(forecast1.member, self.member)
61-
assert_impact_kwargs(forecast1, **impact_kwargs())
66+
assert_impact_kwargs(forecast1, **impact_kwargs)
6267

63-
def test_impact_forecast_from_impact(self, impact):
68+
def test_impact_forecast_from_impact(self, impact, impact_kwargs):
6469
forecast = ImpactForecast.from_impact(
6570
impact, lead_time=self.lead_time, member=self.member
6671
)
6772
npt.assert_array_equal(forecast.lead_time, self.lead_time)
6873
npt.assert_array_equal(forecast.member, self.member)
69-
assert_impact_kwargs(forecast, **impact_kwargs())
74+
assert_impact_kwargs(forecast, **impact_kwargs)

0 commit comments

Comments
 (0)