diff --git a/src/otoole/results/result_package.py b/src/otoole/results/result_package.py index 6917781..9a961cb 100644 --- a/src/otoole/results/result_package.py +++ b/src/otoole/results/result_package.py @@ -364,7 +364,7 @@ def capital_investment_storage(self) -> pd.DataFrame: """ try: capital_cost_storage = self["CapitalCostStorage"] - new_capacity_storage = self["NewCapacityStorage"] + new_capacity_storage = self["NewStorageCapacity"] except KeyError as ex: raise KeyError(self._msg("CapitalInvestmentStorage", str(ex))) diff --git a/tests/results/test_results_package.py b/tests/results/test_results_package.py index 9dc85b3..d3e7619 100644 --- a/tests/results/test_results_package.py +++ b/tests/results/test_results_package.py @@ -372,7 +372,7 @@ def capital_cost_storage(): @fixture -def new_capacity_storage(): +def new_storage_capacity(): df = pd.DataFrame( data=[ ["SIMPLICITY", "DAM", 2014, 1.3], @@ -977,14 +977,14 @@ def test_null(self, null: ResultsPackage): class TestCapitalInvestmentStorage: def test_capital_investment_storage( - self, region, year, capital_cost_storage, new_capacity_storage + self, region, year, capital_cost_storage, new_storage_capacity ): results = { "REGION": region, "YEAR": year, "CapitalCostStorage": capital_cost_storage, - "NewCapacityStorage": new_capacity_storage, + "NewStorageCapacity": new_storage_capacity, } package = ResultsPackage(results)