import pandas as pd
from pandas.testing import assert_series_equal
from powersimdata.design.generation.cost_curves import (
build_supply_curve,
get_supply_data,
ks_test,
lower_bound_index,
)
from powersimdata.tests.mock_grid import MockGrid
mock_plant = {
"plant_id": range(20),
"type": [
"coal",
"coal",
"coal",
"ng",
"coal",
"ng",
"ng",
"ng",
"ng",
"ng",
"coal",
"coal",
"coal",
"coal",
"coal",
"coal",
"coal",
"ng",
"ng",
"ng",
],
"Pmin": [0] * 20,
"Pmax": [
50,
20,
100,
10,
10,
10,
100,
20,
20,
50,
100,
10,
10,
10,
20,
50,
50,
20,
20,
100,
],
"interconnect": ["Western"] * 20,
"zone_name": ["Utah"] * 4 + ["Colorado"] * 6 + ["Washington"] * 10,
}
mock_gencost = {
"plant_id": range(20),
"type": [2] * 20,
"startup": [0] * 20,
"shutdown": [0] * 20,
"n": [3] * 20,
"c2": [
0.025,
0.010,
0.050,
0.010,
0.020,
0.010,
0.050,
0.020,
0.020,
0.025,
0.025,
0.020,
0.010,
0.010,
0.020,
0.020,
0.025,
0.020,
0.010,
0.050,
],
"c1": [
30,
30,
35,
30,
25,
25,
35,
30,
30,
30,
35,
25,
25,
30,
30,
30,
30,
25,
25,
35,
],
"c0": [
1300,
1100,
1700,
1100,
1000,
1200,
1900,
1200,
1100,
1400,
1800,
1100,
1100,
1200,
1400,
1500,
1300,
1100,
1200,
1900,
],
"interconnect": ["Western"] * 20,
}
grid_attrs = {"plant": mock_plant, "gencost_before": mock_gencost}
grid = MockGrid(grid_attrs)
grid.interconnect = "Western"
grid.zone2id = {"Utah": 210, "Colorado": 212, "Washington": 201}
[docs]def test_get_supply_data():
supply_df = get_supply_data(grid, 1)
test_slope = supply_df["slope1"]
exp_slope = pd.Series(
[
31.25,
30.20,
40.00,
30.10,
25.20,
25.10,
40.00,
30.40,
30.40,
31.25,
37.50,
25.20,
25.10,
30.10,
30.40,
31.00,
31.25,
25.40,
25.20,
40.00,
],
index=supply_df.index,
name="slope1",
)
assert_series_equal(test_slope, exp_slope)
[docs]def test_build_supply_curve_1seg():
capacity_test, price_test = build_supply_curve(
grid, 1, "Colorado", "ng", "loadzone", plot=False
)
capacity_exp = [0, 10, 10, 30, 30, 50, 50, 100, 100, 200]
price_exp = [
25.10,
25.10,
30.40,
30.40,
30.40,
30.40,
31.25,
31.25,
40.00,
40.00,
]
assert all([capacity_test[i] == capacity_exp[i] for i in range(len(capacity_test))])
assert all([price_test[i] == price_exp[i] for i in range(len(capacity_test))])
[docs]def test_build_supply_curve_2seg():
capacity_test, price_test = build_supply_curve(
grid, 2, "Utah", "coal", "loadzone", plot=False
)
capacity_exp = [0, 10, 10, 20, 20, 45, 45, 70, 70, 120, 120, 170]
price_exp = [
30.100,
30.100,
30.300,
30.300,
30.625,
30.625,
31.875,
31.875,
37.500,
37.500,
42.500,
42.500,
]
assert all([capacity_test[i] == capacity_exp[i] for i in range(len(capacity_test))])
assert all([price_test[i] == price_exp[i] for i in range(len(capacity_test))])
[docs]def test_ks_test():
capacity_data1, price_data1 = build_supply_curve(
grid, 1, "Washington", "coal", "loadzone", plot=False
)
capacity_data2 = [
0,
15,
15,
40,
40,
75,
75,
130,
130,
190,
190,
225,
225,
max(capacity_data1),
]
price_data2 = [
23.00,
23.00,
27.00,
27.00,
29.00,
29.00,
30.00,
30.00,
33.00,
33.00,
34.00,
34.00,
38.00,
38.00,
]
test_diff = ks_test(
capacity_data1, price_data1, capacity_data2, price_data2, plot=False
)
exp_diff = 4.5
assert test_diff == exp_diff
[docs]def test_lower_bound_index():
desired_capacity = 10
capacity_data = [0, 5, 5, 9, 9, 12, 12, 18]
ind_test = lower_bound_index(desired_capacity, capacity_data)
ind_exp = 4
assert ind_test == ind_exp
