correct NewStorageCapacity name

OSeMOSYS · Nov 8, 2024 · 941bbf2 · 941bbf2
1 parent 7a686b9
commit 941bbf2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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)