Reformat source code

src/lmp/aelmp.jl

@@ -111,14 +111,13 @@ function compute_lmp(
     return elmp
 end
 
-function _aelmp_check_parameters(
-    method::AELMP,
-    model::JuMP.Model
-)
+function _aelmp_check_parameters(method::AELMP, model::JuMP.Model)
     # CHECK: model must be solved if allow_offline_participation=false
     if !method.allow_offline_participation
         if isnothing(model) || !has_values(model)
-            error("A solved UC model is required if allow_offline_participation=false.")
+            error(
+                "A solved UC model is required if allow_offline_participation=false.",
+            )
         end
     end
 end
@@ -126,7 +125,7 @@ end
 function _modify_instance!(
     instance::UnitCommitmentInstance,
     model::JuMP.Model,
-    method::AELMP
+    method::AELMP,
 )
     # this function modifies the instance units (generators)
     # 1. remove (if NOT allowing) the offline generators
@@ -152,10 +151,14 @@ function _modify_instance!(
         # min_power & min_costs are vectors with dimension T
         if unit.min_power[1] != 0
             first_cost_segment = unit.cost_segments[1]
-            pushfirst!(unit.cost_segments, CostSegment(
+            pushfirst!(
+                unit.cost_segments,
+                CostSegment(
                     ones(size(first_cost_segment.mw)) * unit.min_power[1],
-                ones(size(first_cost_segment.cost)) * unit.min_power_cost[1] / unit.min_power[1]
-            ))
+                    ones(size(first_cost_segment.cost)) *
+                    unit.min_power_cost[1] / unit.min_power[1],
+                ),
+            )
             unit.min_power = zeros(size(first_cost_segment.mw))
             unit.min_power_cost = zeros(size(first_cost_segment.cost))
         end
@@ -175,12 +178,13 @@ function _modify_instance!(
         # 4. other adjustments...
         ### FIXME in the future
         # MISO Phase I: can ONLY solve fast-starts, force all startup time to be 0
-        unit.startup_categories = StartupCategory[StartupCategory(0, first_startup_cost)]
+        unit.startup_categories =
+            StartupCategory[StartupCategory(0, first_startup_cost)]
         unit.initial_status = -100
         unit.initial_power = 0
         unit.min_uptime = 0
         unit.min_downtime = 0
         ### END FIXME
     end
-    instance.units_by_name = Dict(g.name => g for g in instance.units)
+    return instance.units_by_name = Dict(g.name => g for g in instance.units)
 end

test/lmp/aelmp_test.jl

@@ -3,8 +3,7 @@
 # Released under the modified BSD license. See COPYING.md for more details.
 
 using UnitCommitment, Cbc, HiGHS, JuMP
-import UnitCommitment:
-    AELMP
+import UnitCommitment: AELMP
 
 @testset "aelmp" begin
     path = "$FIXTURES/aelmp_simple.json.gz"
@@ -19,11 +18,8 @@ import UnitCommitment:
     UnitCommitment.optimize!(model)
 
     # policy 1: allow offlines; consider startups
-    aelmp_1 = UnitCommitment.compute_lmp(
-        model, 
-        AELMP(),
-        optimizer=HiGHS.Optimizer
-    )
+    aelmp_1 =
+        UnitCommitment.compute_lmp(model, AELMP(), optimizer = HiGHS.Optimizer)
     @test aelmp_1["B1", 1] ≈ 231.7 atol = 0.1
 
     # policy 2: do not allow offlines; but consider startups
@@ -31,9 +27,9 @@ import UnitCommitment:
         model,
         AELMP(
             allow_offline_participation = false,
-            consider_startup_costs=true
+            consider_startup_costs = true,
         ),
-        optimizer=HiGHS.Optimizer
+        optimizer = HiGHS.Optimizer,
     )
     @test aelmp_2["B1", 1] ≈ 274.3 atol = 0.1
 end