updated

psrenergy · Nov 23, 2023 · 4a7ab7c · 4a7ab7c
1 parent 81c1180
commit 4a7ab7c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/test/read_and_write_hourly.jl b/test/read_and_write_hourly.jl
@@ -3,7 +3,7 @@ function test_read_and_write_hourly()
 
     stages = 3
     scenarios = 2
-    AGENTS = ["X", "Y", "Z"]
+    agents = ["X", "Y", "Z"]
     unit = "MW"
     stage_type = PSRI.STAGE_MONTH
     initial_stage = 2
@@ -15,7 +15,7 @@ function test_read_and_write_hourly()
         is_hourly = true,
         scenarios = scenarios,
         stages = stages,
-        agents = AGENTS,
+        agents = agents,
         unit = unit,
         # optional:
         stage_type = stage_type,
@@ -55,7 +55,7 @@ function test_read_and_write_hourly()
     @test PSRI.initial_stage(ior) == initial_stage
     @test PSRI.initial_year(ior) == initial_year
     @test PSRI.data_unit(ior) == unit
-    @test PSRI.agent_names(ior) == ["X", "Y", "Z"]
+    @test PSRI.agent_names(ior) == agents
 
     for stage in 1:stages
         for scenario in 1:scenarios

diff --git a/test/read_and_write_monthly.jl b/test/read_and_write_monthly.jl
@@ -4,6 +4,7 @@ function test_read_and_write_monthly()
     stages = 12
     blocks = 3
     scenarios = 4
+    agents = ["X", "Y", "Z"]
     stage_type = PSRI.STAGE_MONTH
     initial_stage = 1
     initial_year = 2006
@@ -15,7 +16,7 @@ function test_read_and_write_monthly()
         blocks = blocks,
         scenarios = scenarios,
         stages = stages,
-        agents = ["X", "Y", "Z"],
+        agents = agents,
         unit = unit,
         # optional:
         stage_type = stage_type,
@@ -50,8 +51,7 @@ function test_read_and_write_monthly()
     @test PSRI.initial_stage(ior) == initial_stage
     @test PSRI.initial_year(ior) == initial_year
     @test PSRI.data_unit(ior) == unit
-
-    @test PSRI.agent_names(ior) == ["X", "Y", "Z"]
+    @test PSRI.agent_names(ior) == agents
 
     for stage in 1:stages
         for scenario in 1:scenarios