# UnitCommitment.jl: Optimization Package for Security-Constrained Unit Commitment
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
# Released under the modified BSD license. See COPYING.md for more details.

using UnitCommitment, LinearAlgebra, Cbc, JuMP, JSON, GZip

@testset "read_benchmark" begin
    instance = UnitCommitment.read("$FIXTURES/case14.json.gz")

    @test repr(instance) == (
        "UnitCommitmentInstance(1 scenarios, 6 units, 14 buses, " *
        "20 lines, 19 contingencies, 1 price sensitive loads, 0 profiled units, 4 time steps)"
    )

    @test length(instance.scenarios) == 1
    sc = instance.scenarios[1]
    @test length(sc.lines) == 20
    @test length(sc.buses) == 14
    @test length(sc.units) == 6
    @test length(sc.contingencies) == 19
    @test length(sc.price_sensitive_loads) == 1
    @test instance.time == 4

    @test sc.lines[5].name == "l5"
    @test sc.lines[5].source.name == "b2"
    @test sc.lines[5].target.name == "b5"
    @test sc.lines[5].reactance ≈ 0.17388
    @test sc.lines[5].susceptance ≈ 10.037550333
    @test sc.lines[5].normal_flow_limit == [1e8 for t in 1:4]
    @test sc.lines[5].emergency_flow_limit == [1e8 for t in 1:4]
    @test sc.lines[5].flow_limit_penalty == [5e3 for t in 1:4]
    @test sc.lines_by_name["l5"].name == "l5"

    @test sc.lines[1].name == "l1"
    @test sc.lines[1].source.name == "b1"
    @test sc.lines[1].target.name == "b2"
    @test sc.lines[1].reactance ≈ 0.059170
    @test sc.lines[1].susceptance ≈ 29.496860773945
    @test sc.lines[1].normal_flow_limit == [300.0 for t in 1:4]
    @test sc.lines[1].emergency_flow_limit == [400.0 for t in 1:4]
    @test sc.lines[1].flow_limit_penalty == [1e3 for t in 1:4]

    @test sc.buses[9].name == "b9"
    @test sc.buses[9].load == [35.36638, 33.25495, 31.67138, 31.14353]
    @test sc.buses_by_name["b9"].name == "b9"

    @test sc.reserves[1].name == "r1"
    @test sc.reserves[1].type == "spinning"
    @test sc.reserves[1].amount == [100.0, 100.0, 100.0, 100.0]
    @test sc.reserves_by_name["r1"].name == "r1"

    unit = sc.units[1]
    @test unit.name == "g1"
    @test unit.bus.name == "b1"
    @test unit.ramp_up_limit == 1e6
    @test unit.ramp_down_limit == 1e6
    @test unit.startup_limit == 1e6
    @test unit.shutdown_limit == 1e6
    @test unit.must_run == [false for t in 1:4]
    @test unit.min_power_cost == [1400.0 for t in 1:4]
    @test unit.min_uptime == 1
    @test unit.min_downtime == 1
    for t in 1:1
        @test unit.cost_segments[1].mw[t] == 10.0
        @test unit.cost_segments[2].mw[t] == 20.0
        @test unit.cost_segments[3].mw[t] == 5.0
        @test unit.cost_segments[1].cost[t] ≈ 20.0
        @test unit.cost_segments[2].cost[t] ≈ 30.0
        @test unit.cost_segments[3].cost[t] ≈ 40.0
    end
    @test length(unit.startup_categories) == 3
    @test unit.startup_categories[1].delay == 1
    @test unit.startup_categories[2].delay == 2
    @test unit.startup_categories[3].delay == 3
    @test unit.startup_categories[1].cost == 1000.0
    @test unit.startup_categories[2].cost == 1500.0
    @test unit.startup_categories[3].cost == 2000.0
    @test length(unit.reserves) == 0
    @test sc.units_by_name["g1"].name == "g1"

    unit = sc.units[2]
    @test unit.name == "g2"
    @test unit.must_run == [false for t in 1:4]
    @test length(unit.reserves) == 1

    unit = sc.units[3]
    @test unit.name == "g3"
    @test unit.bus.name == "b3"
    @test unit.ramp_up_limit == 70.0
    @test unit.ramp_down_limit == 70.0
    @test unit.startup_limit == 70.0
    @test unit.shutdown_limit == 70.0
    @test unit.must_run == [true for t in 1:4]
    @test unit.min_power_cost == [0.0 for t in 1:4]
    @test unit.min_uptime == 1
    @test unit.min_downtime == 1
    for t in 1:4
        @test unit.cost_segments[1].mw[t] ≈ 33
        @test unit.cost_segments[2].mw[t] ≈ 33
        @test unit.cost_segments[3].mw[t] ≈ 34
        @test unit.cost_segments[1].cost[t] ≈ 33.75
        @test unit.cost_segments[2].cost[t] ≈ 38.04
        @test unit.cost_segments[3].cost[t] ≈ 44.77853
    end
    @test length(unit.reserves) == 1
    @test unit.reserves[1].name == "r1"

    @test sc.contingencies[1].lines == [sc.lines[1]]
    @test sc.contingencies[1].units == []
    @test sc.contingencies[1].name == "c1"
    @test sc.contingencies_by_name["c1"].name == "c1"

    load = sc.price_sensitive_loads[1]
    @test load.name == "ps1"
    @test load.bus.name == "b3"
    @test load.revenue == [100.0 for t in 1:4]
    @test load.demand == [50.0 for t in 1:4]
    @test sc.price_sensitive_loads_by_name["ps1"].name == "ps1"
end

@testset "read_benchmark sub-hourly" begin
    instance = UnitCommitment.read("$FIXTURES/case14-sub-hourly.json.gz")
    @test instance.time == 4
    unit = instance.scenarios[1].units[1]
    @test unit.name == "g1"
    @test unit.min_uptime == 2
    @test unit.min_downtime == 2
    @test length(unit.startup_categories) == 3
    @test unit.startup_categories[1].delay == 2
    @test unit.startup_categories[2].delay == 4
    @test unit.startup_categories[3].delay == 6
    @test unit.initial_status == -200
end

@testset "read_benchmark profiled-units" begin
    instance = UnitCommitment.read("$FIXTURES/case14-profiled.json.gz")
    sc = instance.scenarios[1]
    @test length(sc.profiled_units) == 2

    first_pu = sc.profiled_units[1]
    @test first_pu.name == "g7"
    @test first_pu.bus.name == "b4"
    @test first_pu.cost == [100.0 for t in 1:4]
    @test first_pu.capacity == [100.0 for t in 1:4]
    @test sc.profiled_units_by_name["g7"].name == "g7"

    second_pu = sc.profiled_units[2]
    @test second_pu.name == "g8"
    @test second_pu.bus.name == "b5"
    @test second_pu.cost == [50.0 for t in 1:4]
    @test second_pu.capacity == [120.0 for t in 1:4]
    @test sc.profiled_units_by_name["g8"].name == "g8"
end