def test_type_parsers(self):
for case in test_cases.CASES:
if case.split('.')[-1] == 'mat':
self.assertRaises(pf.ParserError,pf.ParserJSON().parse,case)
self.assertRaises(pf.ParserError,pf.PyParserMAT().parse,case)
if pf.has_raw_parser():
self.assertRaises(pf.ParserError,pf.ParserRAW().parse,case)
net = pf.ParserMAT().parse(case)
self.assertGreater(net.num_buses,0)
if case.split('.')[-1] == 'm':
self.assertRaises(pf.ParserError,pf.ParserJSON().parse,case)
self.assertRaises(pf.ParserError,pf.ParserMAT().parse,case)
if pf.has_raw_parser():
self.assertRaises(pf.ParserError,pf.ParserRAW().parse,case)
net = pf.PyParserMAT().parse(case)
self.assertGreater(net.num_buses,0)
def test_ieee14_gen_cost(self):
for case in test_cases.CASES:
if case == './data/ieee14.mat':
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power,100.)
self.assertEqual(net.num_buses,14)
self.assertEqual(net.num_generators,5)
gen0 = net.get_generator(net.num_generators-1)
gen1 = net.get_generator(net.num_generators-2)
self.assertEqual(gen0.P,232.4/100.)
self.assertEqual(gen0.cost_coeff_Q2,(4.3029259900e-02)*(net.base_power**2.))
self.assertEqual(gen0.cost_coeff_Q1,20.*net.base_power)
self.assertEqual(gen1.P,40./100.)
self.assertEqual(gen1.cost_coeff_Q2,0.25*(net.base_power**2.))
self.assertEqual(gen1.cost_coeff_Q1,20.*net.base_power)
self.assertEqual(branch.name, '1')
self.assertEqual(len(brs), 2)
self.assertEqual([br.name for br in brs], ['2', '1'])
self.assertTrue(tested_101)
self.assertTrue(tested_104)
self.assertTrue(tested_106)
self.assertTrue(tested_222)
def test_sys_problem2(self):
case = os.path.join('data', 'sys_problem2.mat')
if not os.path.isfile(case):
raise unittest.SkipTest('file not available')
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power, 100.)
self.assertEqual(net.num_buses, 3)
self.assertEqual(net.num_generators, 4)
self.assertEqual(net.num_loads, 3)
self.assertEqual(net.num_branches, 3)
bus1 = net.get_bus_from_number(1)
bus2 = net.get_bus_from_number(2)
bus3 = net.get_bus_from_number(3)
for bus in net.buses:
self.assertEqual(bus.v_base, 220.)
def test_sys_problem2(self):
for case in test_cases.CASES:
if case == './data/sys_problem2.mat':
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power, 100.)
self.assertEqual(net.num_buses, 3)
self.assertEqual(net.num_generators, 4)
self.assertEqual(net.num_loads, 3)
self.assertEqual(net.num_branches, 3)
bus1 = net.get_bus_from_number(1)
bus2 = net.get_bus_from_number(2)
bus3 = net.get_bus_from_number(3)
for bus in net.buses:
self.assertEqual(bus.v_base, 220.)
self.assertEqual(bus1.number, 1)
self.assertEqual(bus2.number, 2)
self.assertEqual(bus3.number, 3)
branch13 = net.get_branch(0)
branch23 = net.get_branch(1)
branch12 = net.get_branch(2)
self.assertEqual(len(bus1.generators), 2)
self.assertEqual(len(bus2.generators), 1)
self.assertEqual(len(bus3.generators), 1)
self.assertEqual(len(bus1.loads), 1)
self.assertEqual(len(bus2.loads), 1)
self.assertEqual(len(bus3.loads), 1)
gen0 = net.get_generator(3)
gen1 = net.get_generator(2)
gen2 = net.get_generator(1)
gen3 = net.get_generator(0)
load0 = net.get_load(2)
load1 = net.get_load(1)
load2 = net.get_load(0)
self.assertEqual(load0.bus, bus1)
self.assertEqual(load1.bus, bus2)
self.assertEqual(load2.bus, bus3)
self.assertEqual(gen0.P, 50 / 100.)
self.assertEqual(gen1.P, 40 / 100.)
self.assertEqual(gen2.P, 30 / 100.)
self.assertEqual(gen3.P, 20 / 100.)
self.assertEqual(gen0.P_max, 50)
self.assertEqual(gen1.P_max, 50)
self.assertEqual(gen2.P_max, 25)
self.assertEqual(gen3.P_max, 19)
self.assertEqual(gen0.bus, bus1)
self.assertEqual(gen1.bus, bus1)
self.assertEqual(gen2.bus, bus2)
self.assertEqual(gen3.bus, bus3)
for gen in net.generators:
self.assertEqual(gen.P_min, 0.)
self.assertEqual(branch13.bus_k.number, bus1.number)
self.assertEqual(branch13.bus_m.number, bus3.number)
self.assertEqual(branch23.bus_k.number, bus2.number)
self.assertEqual(branch23.bus_m.number, bus3.number)
self.assertEqual(branch12.bus_k.number, bus1.number)
self.assertEqual(branch12.bus_m.number, bus2.number)
self.assertEqual(branch13.g, 0)
self.assertLess(abs(branch13.b + 1. / 0.1), 1e-10)
self.assertEqual(branch13.ratingA, 30.95)
self.assertEqual(branch13.ratingB, 30.95)
self.assertEqual(branch13.ratingC, 30.95)
self.assertEqual(branch23.g, 0)
self.assertLess(abs(branch23.b + 1. / 0.2), 1e-10)
self.assertEqual(branch23.ratingA, 13)
self.assertEqual(branch23.ratingB, 13)
self.assertEqual(branch23.ratingC, 13)
self.assertEqual(branch12.g, 0)
self.assertLess(abs(branch12.b + 1. / 0.2), 1e-10)
self.assertEqual(branch12.ratingA, 15)
self.assertEqual(branch12.ratingB, 15)
self.assertEqual(branch12.ratingC, 15)
self.assertEqual(gen0.cost_coeff_Q0, 0)
self.assertEqual(gen0.cost_coeff_Q1, 6. * net.base_power)
self.assertEqual(gen0.cost_coeff_Q2,
0.03 * (net.base_power**2.))
self.assertEqual(gen1.cost_coeff_Q0, 0)
self.assertEqual(gen1.cost_coeff_Q1, 5. * net.base_power)
self.assertEqual(gen1.cost_coeff_Q2,
0.02 * (net.base_power**2.))
self.assertEqual(gen2.cost_coeff_Q0, 0)
self.assertEqual(gen2.cost_coeff_Q1, 12. * net.base_power)
self.assertEqual(gen2.cost_coeff_Q2,
0.06 * (net.base_power**2.))
self.assertEqual(gen3.cost_coeff_Q0, 0)
self.assertEqual(gen3.cost_coeff_Q1, 10. * net.base_power)
self.assertEqual(gen3.cost_coeff_Q2,
0.08 * (net.base_power**2.))
# Load utility
self.assertEqual(load0.util_coeff_Q0, 0)
self.assertEqual(load0.util_coeff_Q1, 400. * net.base_power)
self.assertEqual(load0.util_coeff_Q2,
-0.03 * (net.base_power**2.))
self.assertEqual(load1.util_coeff_Q0, 0)
self.assertEqual(load1.util_coeff_Q1, 450. * net.base_power)
self.assertEqual(load1.util_coeff_Q2,
-0.02 * (net.base_power**2.))
self.assertEqual(load2.util_coeff_Q0, 0)
self.assertEqual(load2.util_coeff_Q1, 300. * net.base_power)
self.assertEqual(load2.util_coeff_Q2,
-0.03 * (net.base_power**2.))
def test_sys_problem3(self):
for case in test_cases.CASES:
if case == './data/sys_problem3.mat':
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power, 100.)
# numbers
self.assertEqual(net.num_buses, 10)
self.assertEqual(net.num_generators, 5)
self.assertEqual(net.num_loads, 10)
self.assertEqual(net.num_shunts, 0)
self.assertEqual(net.num_var_generators, 0)
self.assertEqual(net.num_branches, 13)
# buses
bus1 = net.get_bus_from_number(1)
bus2 = net.get_bus_from_number(2)
bus3 = net.get_bus_from_number(3)
bus4 = net.get_bus_from_number(4)
bus5 = net.get_bus_from_number(5)
bus6 = net.get_bus_from_number(6)
bus7 = net.get_bus_from_number(7)
bus8 = net.get_bus_from_number(8)
bus9 = net.get_bus_from_number(9)
bus10 = net.get_bus_from_number(10)
for bus in net.buses:
self.assertEqual(bus.v_base, 69.)
# loads
for bus in net.buses:
self.assertEqual(len(bus.loads), 1)
load1 = bus1.loads[0]
load2 = bus2.loads[0]
load3 = bus3.loads[0]
load4 = bus4.loads[0]
load5 = bus5.loads[0]
load6 = bus6.loads[0]
load7 = bus7.loads[0]
load8 = bus8.loads[0]
load9 = bus9.loads[0]
load10 = bus10.loads[0]
self.assertEqual(load1.bus, bus1)
self.assertEqual(load2.bus, bus2)
self.assertEqual(load3.bus, bus3)
self.assertEqual(load4.bus, bus4)
self.assertEqual(load5.bus, bus5)
self.assertEqual(load6.bus, bus6)
self.assertEqual(load7.bus, bus7)
self.assertEqual(load8.bus, bus8)
self.assertEqual(load9.bus, bus9)
self.assertEqual(load10.bus, bus10)
self.assertEqual(load1.P, 55. / 100.)
self.assertEqual(load2.P, 55 / 100.)
self.assertEqual(load3.P, 1300 / 100.)
self.assertEqual(load4.P, 650 / 100.)
self.assertEqual(load5.P, 650 / 100.)
self.assertEqual(load6.P, 200 / 100.)
self.assertEqual(load7.P, 2600 / 100.)
self.assertEqual(load8.P, 3600 / 100.)
self.assertEqual(load9.P, 1100 / 100.)
self.assertEqual(load10.P, 1900 / 100.)
for load in net.loads:
self.assertEqual(load.P_max, load.P)
self.assertEqual(load.P_min, load.P)
# generators
self.assertEqual(len(bus1.generators), 0)
self.assertEqual(len(bus2.generators), 1)
self.assertEqual(len(bus3.generators), 1)
self.assertEqual(len(bus4.generators), 0)
self.assertEqual(len(bus5.generators), 1)
self.assertEqual(len(bus6.generators), 0)
self.assertEqual(len(bus7.generators), 1)
self.assertEqual(len(bus8.generators), 1)
self.assertEqual(len(bus9.generators), 0)
self.assertEqual(len(bus10.generators), 0)
gen1 = bus2.generators[0]
gen2 = bus3.generators[0]
gen3 = bus5.generators[0]
gen4 = bus7.generators[0]
gen5 = bus8.generators[0]
self.assertEqual(gen1.bus, bus2)
self.assertEqual(gen2.bus, bus3)
self.assertEqual(gen3.bus, bus5)
self.assertEqual(gen4.bus, bus7)
self.assertEqual(gen5.bus, bus8)
self.assertEqual(gen1.P, 50. / 100.)
self.assertEqual(gen1.P_min, 0)
self.assertEqual(gen1.P_max, 1200. / 100.)
self.assertEqual(gen1.cost_coeff_Q0, 0)
self.assertEqual(gen1.cost_coeff_Q1, 6.9 * 100)
self.assertEqual(gen1.cost_coeff_Q2, 0.00067 * (100**2.))
self.assertEqual(gen2.P, 40. / 100.)
self.assertEqual(gen2.P_min, 0)
self.assertEqual(gen2.P_max, 8000. / 100.)
self.assertEqual(gen2.cost_coeff_Q0, 0)
self.assertEqual(gen2.cost_coeff_Q1, 24.3 * 100)
self.assertEqual(gen2.cost_coeff_Q2, 0.00040 * (100**2.))
self.assertEqual(gen3.P, 30. / 100.)
self.assertEqual(gen3.P_min, 0)
self.assertEqual(gen3.P_max, 3000. / 100.)
self.assertEqual(gen3.cost_coeff_Q0, 0)
self.assertEqual(gen3.cost_coeff_Q1, 29.1 * 100)
self.assertEqual(gen3.cost_coeff_Q2, 0.00006 * (100**2.))
self.assertEqual(gen4.P, 20. / 100.)
self.assertEqual(gen4.P_min, 0)
self.assertEqual(gen4.P_max, 800. / 100.)
self.assertEqual(gen4.cost_coeff_Q0, 0)
self.assertEqual(gen4.cost_coeff_Q1, 6.9 * 100)
self.assertEqual(gen4.cost_coeff_Q2, 0.00026 * (100**2.))
self.assertEqual(gen5.P, 10. / 100.)
self.assertEqual(gen5.P_min, 0)
self.assertEqual(gen5.P_max, 2000. / 100.)
self.assertEqual(gen5.cost_coeff_Q0, 0)
self.assertEqual(gen5.cost_coeff_Q1, 50. * 100)
self.assertEqual(gen5.cost_coeff_Q2, 0.0015 * (100**2.))
# branches
branch1 = net.get_branch(12)
branch2 = net.get_branch(11)
branch3 = net.get_branch(10)
branch4 = net.get_branch(9)
branch5 = net.get_branch(8)
branch6 = net.get_branch(7)
branch7 = net.get_branch(6)
branch8 = net.get_branch(5)
branch9 = net.get_branch(4)
branch10 = net.get_branch(3)
branch11 = net.get_branch(2)
branch12 = net.get_branch(1)
branch13 = net.get_branch(0)
self.assertEqual(branch1.bus_k, bus1)
self.assertEqual(branch1.bus_m, bus3)
self.assertLess(abs(branch1.b + 1. / 0.1), 1e-10)
self.assertEqual(branch1.ratingA, 3000. / 100.)
self.assertEqual(branch1.ratingB, 3000. / 100.)
self.assertEqual(branch1.ratingC, 3000. / 100.)
self.assertEqual(branch2.bus_k, bus1)
self.assertEqual(branch2.bus_m, bus10)
self.assertLess(abs(branch2.b + 1. / 0.27), 1e-10)
self.assertEqual(branch2.ratingA, 2000. / 100.)
self.assertEqual(branch2.ratingB, 2000. / 100.)
self.assertEqual(branch2.ratingC, 2000. / 100.)
self.assertEqual(branch3.bus_k, bus2)
self.assertEqual(branch3.bus_m, bus3)
self.assertLess(abs(branch3.b + 1. / 0.12), 1e-10)
self.assertEqual(branch3.ratingA, 6500. / 100.)
self.assertEqual(branch3.ratingB, 6500. / 100.)
self.assertEqual(branch3.ratingC, 6500. / 100.)
self.assertEqual(branch4.bus_k, bus2)
self.assertEqual(branch4.bus_m, bus9)
self.assertLess(abs(branch4.b + 1. / 0.07), 1e-10)
self.assertEqual(branch4.ratingA, 5500. / 100.)
self.assertEqual(branch4.ratingB, 5500. / 100.)
self.assertEqual(branch4.ratingC, 5500. / 100.)
self.assertEqual(branch5.bus_k, bus2)
self.assertEqual(branch5.bus_m, bus10)
self.assertLess(abs(branch5.b + 1. / 0.14), 1e-10)
self.assertEqual(branch5.ratingA, 5500. / 100.)
self.assertEqual(branch5.ratingB, 5500. / 100.)
self.assertEqual(branch5.ratingC, 5500. / 100.)
self.assertEqual(branch6.bus_k, bus3)
self.assertEqual(branch6.bus_m, bus4)
self.assertLess(abs(branch6.b + 1. / 0.1), 1e-10)
self.assertEqual(branch6.ratingA, 3000. / 100.)
self.assertEqual(branch6.ratingB, 3000. / 100.)
self.assertEqual(branch6.ratingC, 3000. / 100.)
self.assertEqual(branch7.bus_k, bus3)
self.assertEqual(branch7.bus_m, bus5)
self.assertLess(abs(branch7.b + 1. / 0.17), 1e-10)
self.assertEqual(branch7.ratingA, 4000. / 100.)
self.assertEqual(branch7.ratingB, 4000. / 100.)
self.assertEqual(branch7.ratingC, 4000. / 100.)
self.assertEqual(branch8.bus_k, bus4)
self.assertEqual(branch8.bus_m, bus5)
self.assertLess(abs(branch8.b + 1. / 0.17), 1e-10)
self.assertEqual(branch8.ratingA, 4000. / 100.)
self.assertEqual(branch8.ratingB, 4000. / 100.)
self.assertEqual(branch8.ratingC, 4000. / 100.)
self.assertEqual(branch9.bus_k, bus5)
self.assertEqual(branch9.bus_m, bus6)
self.assertLess(abs(branch9.b + 1. / 0.17), 1e-10)
self.assertEqual(branch9.ratingA, 5000. / 100.)
self.assertEqual(branch9.ratingB, 5000. / 100.)
self.assertEqual(branch9.ratingC, 5000. / 100.)
self.assertEqual(branch10.bus_k, bus6)
self.assertEqual(branch10.bus_m, bus7)
self.assertLess(abs(branch10.b + 1. / 0.16), 1e-10)
self.assertEqual(branch10.ratingA, 2000. / 100.)
self.assertEqual(branch10.ratingB, 2000. / 100.)
self.assertEqual(branch10.ratingC, 2000. / 100.)
self.assertEqual(branch11.bus_k, bus7)
self.assertEqual(branch11.bus_m, bus8)
self.assertLess(abs(branch11.b + 1. / 0.25), 1e-10)
self.assertEqual(branch11.ratingA, 3000. / 100.)
self.assertEqual(branch11.ratingB, 3000. / 100.)
self.assertEqual(branch11.ratingC, 3000. / 100.)
self.assertEqual(branch12.bus_k, bus8)
self.assertEqual(branch12.bus_m, bus9)
self.assertLess(abs(branch12.b + 1. / 0.25), 1e-10)
self.assertEqual(branch12.ratingA, 2500. / 100.)
self.assertEqual(branch12.ratingB, 2500. / 100.)
self.assertEqual(branch12.ratingC, 2500. / 100.)
self.assertEqual(branch13.bus_k, bus8)
self.assertEqual(branch13.bus_m, bus10)
self.assertLess(abs(branch13.b + 1. / 0.07), 1e-10)
self.assertEqual(branch13.ratingA, 4000. / 100.)
self.assertEqual(branch13.ratingB, 4000. / 100.)
self.assertEqual(branch13.ratingC, 4000. / 100.)
class TestParser(unittest.TestCase):
def setUp(self):
pass
def test_ieee25_raw(self):
for case in test_cases.CASES:
if case == './data/ieee25.raw':
net = pf.ParserRAW().parse(case)
self.assertEqual(net.num_buses, 25)
self.assertTrue(101 in [bus.number for bus in net.buses])
self.assertTrue(104 in [bus.number for bus in net.buses])
self.assertTrue(106 in [bus.number for bus in net.buses])
self.assertTrue(222 in [bus.number for bus in net.buses])
tested_101 = False
tested_104 = False
tested_106 = False
tested_222 = False
for bus in net.buses:
# base kv
if bus.number >= 101 and bus.number <= 110:
self.assertEqual(bus.v_base, 138.)
else:
self.assertLessEqual(bus.number, 225)
self.assertGreaterEqual(bus.number, 211)
self.assertEqual(bus.v_base, 230.)
# names
if bus.number == 101:
self.assertEqual(bus.name, 'COAL-A')
self.assertEqual(len(bus.generators), 3)
self.assertEqual([g.name for g in bus.generators],
['5', '4', '3'])
tested_101 = True
if bus.number == 104:
self.assertEqual(len(bus.loads), 1)
self.assertEqual(bus.loads[0].name, '1')
tested_104 = True
if bus.number == 106:
self.assertEqual(len(bus.shunts), 2)
self.assertEqual([s.name for s in bus.shunts],
['', '1'])
tested_106 = True
if bus.number == 222:
self.assertEqual(bus.name, 'HYDRO')
self.assertEqual(len(bus.generators), 10)
self.assertEqual(
[g.name for g in bus.generators],
['A', '9', '8', '7', '6', '5', '4', '3', '2', '1'])
tested_222 = True
brs = []
for branch in net.branches:
if (branch.bus_k.number, branch.bus_m.number) == (215,
221):
brs.append(branch)
if (branch.bus_k.number, branch.bus_m.number) == (103,
224):
self.assertEqual(branch.name, '1')
self.assertEqual(len(brs), 2)
self.assertEqual([br.name for br in brs], ['2', '1'])
self.assertTrue(tested_101)
self.assertTrue(tested_104)
self.assertTrue(tested_106)
self.assertTrue(tested_222)
def test_sys_problem2(self):
for case in test_cases.CASES:
if case == './data/sys_problem2.mat':
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power, 100.)
self.assertEqual(net.num_buses, 3)
self.assertEqual(net.num_generators, 4)
self.assertEqual(net.num_loads, 3)
self.assertEqual(net.num_branches, 3)
bus1 = net.get_bus_from_number(1)
bus2 = net.get_bus_from_number(2)
bus3 = net.get_bus_from_number(3)
for bus in net.buses:
self.assertEqual(bus.v_base, 220.)
self.assertEqual(bus1.number, 1)
self.assertEqual(bus2.number, 2)
self.assertEqual(bus3.number, 3)
branch13 = net.get_branch(0)
branch23 = net.get_branch(1)
branch12 = net.get_branch(2)
self.assertEqual(len(bus1.generators), 2)
self.assertEqual(len(bus2.generators), 1)
self.assertEqual(len(bus3.generators), 1)
self.assertEqual(len(bus1.loads), 1)
self.assertEqual(len(bus2.loads), 1)
self.assertEqual(len(bus3.loads), 1)
gen0 = net.get_generator(3)
gen1 = net.get_generator(2)
gen2 = net.get_generator(1)
gen3 = net.get_generator(0)
load0 = net.get_load(2)
load1 = net.get_load(1)
load2 = net.get_load(0)
self.assertEqual(load0.bus, bus1)
self.assertEqual(load1.bus, bus2)
self.assertEqual(load2.bus, bus3)
self.assertEqual(gen0.P, 50 / 100.)
self.assertEqual(gen1.P, 40 / 100.)
self.assertEqual(gen2.P, 30 / 100.)
self.assertEqual(gen3.P, 20 / 100.)
self.assertEqual(gen0.P_max, 50)
self.assertEqual(gen1.P_max, 50)
self.assertEqual(gen2.P_max, 25)
self.assertEqual(gen3.P_max, 19)
self.assertEqual(gen0.bus, bus1)
self.assertEqual(gen1.bus, bus1)
self.assertEqual(gen2.bus, bus2)
self.assertEqual(gen3.bus, bus3)
for gen in net.generators:
self.assertEqual(gen.P_min, 0.)
self.assertEqual(branch13.bus_k.number, bus1.number)
self.assertEqual(branch13.bus_m.number, bus3.number)
self.assertEqual(branch23.bus_k.number, bus2.number)
self.assertEqual(branch23.bus_m.number, bus3.number)
self.assertEqual(branch12.bus_k.number, bus1.number)
self.assertEqual(branch12.bus_m.number, bus2.number)
self.assertEqual(branch13.g, 0)
self.assertLess(abs(branch13.b + 1. / 0.1), 1e-10)
self.assertEqual(branch13.ratingA, 30.95)
self.assertEqual(branch13.ratingB, 30.95)
self.assertEqual(branch13.ratingC, 30.95)
self.assertEqual(branch23.g, 0)
self.assertLess(abs(branch23.b + 1. / 0.2), 1e-10)
self.assertEqual(branch23.ratingA, 13)
self.assertEqual(branch23.ratingB, 13)
self.assertEqual(branch23.ratingC, 13)
self.assertEqual(branch12.g, 0)
self.assertLess(abs(branch12.b + 1. / 0.2), 1e-10)
self.assertEqual(branch12.ratingA, 15)
self.assertEqual(branch12.ratingB, 15)
self.assertEqual(branch12.ratingC, 15)
self.assertEqual(gen0.cost_coeff_Q0, 0)
self.assertEqual(gen0.cost_coeff_Q1, 6. * net.base_power)
self.assertEqual(gen0.cost_coeff_Q2,
0.03 * (net.base_power**2.))
self.assertEqual(gen1.cost_coeff_Q0, 0)
self.assertEqual(gen1.cost_coeff_Q1, 5. * net.base_power)
self.assertEqual(gen1.cost_coeff_Q2,
0.02 * (net.base_power**2.))
self.assertEqual(gen2.cost_coeff_Q0, 0)
self.assertEqual(gen2.cost_coeff_Q1, 12. * net.base_power)
self.assertEqual(gen2.cost_coeff_Q2,
0.06 * (net.base_power**2.))
self.assertEqual(gen3.cost_coeff_Q0, 0)
self.assertEqual(gen3.cost_coeff_Q1, 10. * net.base_power)
self.assertEqual(gen3.cost_coeff_Q2,
0.08 * (net.base_power**2.))
# Load utility
self.assertEqual(load0.util_coeff_Q0, 0)
self.assertEqual(load0.util_coeff_Q1, 400. * net.base_power)
self.assertEqual(load0.util_coeff_Q2,
-0.03 * (net.base_power**2.))
self.assertEqual(load1.util_coeff_Q0, 0)
self.assertEqual(load1.util_coeff_Q1, 450. * net.base_power)
self.assertEqual(load1.util_coeff_Q2,
-0.02 * (net.base_power**2.))
self.assertEqual(load2.util_coeff_Q0, 0)
self.assertEqual(load2.util_coeff_Q1, 300. * net.base_power)
self.assertEqual(load2.util_coeff_Q2,
-0.03 * (net.base_power**2.))
def test_sys_problem3(self):
for case in test_cases.CASES:
if case == './data/sys_problem3.mat':
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power, 100.)
# numbers
self.assertEqual(net.num_buses, 10)
self.assertEqual(net.num_generators, 5)
self.assertEqual(net.num_loads, 10)
self.assertEqual(net.num_shunts, 0)
self.assertEqual(net.num_var_generators, 0)
self.assertEqual(net.num_branches, 13)
# buses
bus1 = net.get_bus_from_number(1)
bus2 = net.get_bus_from_number(2)
bus3 = net.get_bus_from_number(3)
bus4 = net.get_bus_from_number(4)
bus5 = net.get_bus_from_number(5)
bus6 = net.get_bus_from_number(6)
bus7 = net.get_bus_from_number(7)
bus8 = net.get_bus_from_number(8)
bus9 = net.get_bus_from_number(9)
bus10 = net.get_bus_from_number(10)
for bus in net.buses:
self.assertEqual(bus.v_base, 69.)
# loads
for bus in net.buses:
self.assertEqual(len(bus.loads), 1)
load1 = bus1.loads[0]
load2 = bus2.loads[0]
load3 = bus3.loads[0]
load4 = bus4.loads[0]
load5 = bus5.loads[0]
load6 = bus6.loads[0]
load7 = bus7.loads[0]
load8 = bus8.loads[0]
load9 = bus9.loads[0]
load10 = bus10.loads[0]
self.assertEqual(load1.bus, bus1)
self.assertEqual(load2.bus, bus2)
self.assertEqual(load3.bus, bus3)
self.assertEqual(load4.bus, bus4)
self.assertEqual(load5.bus, bus5)
self.assertEqual(load6.bus, bus6)
self.assertEqual(load7.bus, bus7)
self.assertEqual(load8.bus, bus8)
self.assertEqual(load9.bus, bus9)
self.assertEqual(load10.bus, bus10)
self.assertEqual(load1.P, 55. / 100.)
self.assertEqual(load2.P, 55 / 100.)
self.assertEqual(load3.P, 1300 / 100.)
self.assertEqual(load4.P, 650 / 100.)
self.assertEqual(load5.P, 650 / 100.)
self.assertEqual(load6.P, 200 / 100.)
self.assertEqual(load7.P, 2600 / 100.)
self.assertEqual(load8.P, 3600 / 100.)
self.assertEqual(load9.P, 1100 / 100.)
self.assertEqual(load10.P, 1900 / 100.)
for load in net.loads:
self.assertEqual(load.P_max, load.P)
self.assertEqual(load.P_min, load.P)
# generators
self.assertEqual(len(bus1.generators), 0)
self.assertEqual(len(bus2.generators), 1)
self.assertEqual(len(bus3.generators), 1)
self.assertEqual(len(bus4.generators), 0)
self.assertEqual(len(bus5.generators), 1)
self.assertEqual(len(bus6.generators), 0)
self.assertEqual(len(bus7.generators), 1)
self.assertEqual(len(bus8.generators), 1)
self.assertEqual(len(bus9.generators), 0)
self.assertEqual(len(bus10.generators), 0)
gen1 = bus2.generators[0]
gen2 = bus3.generators[0]
gen3 = bus5.generators[0]
gen4 = bus7.generators[0]
gen5 = bus8.generators[0]
self.assertEqual(gen1.bus, bus2)
self.assertEqual(gen2.bus, bus3)
self.assertEqual(gen3.bus, bus5)
self.assertEqual(gen4.bus, bus7)
self.assertEqual(gen5.bus, bus8)
self.assertEqual(gen1.P, 50. / 100.)
self.assertEqual(gen1.P_min, 0)
self.assertEqual(gen1.P_max, 1200. / 100.)
self.assertEqual(gen1.cost_coeff_Q0, 0)
self.assertEqual(gen1.cost_coeff_Q1, 6.9 * 100)
self.assertEqual(gen1.cost_coeff_Q2, 0.00067 * (100**2.))
self.assertEqual(gen2.P, 40. / 100.)
self.assertEqual(gen2.P_min, 0)
self.assertEqual(gen2.P_max, 8000. / 100.)
self.assertEqual(gen2.cost_coeff_Q0, 0)
self.assertEqual(gen2.cost_coeff_Q1, 24.3 * 100)
self.assertEqual(gen2.cost_coeff_Q2, 0.00040 * (100**2.))
self.assertEqual(gen3.P, 30. / 100.)
self.assertEqual(gen3.P_min, 0)
self.assertEqual(gen3.P_max, 3000. / 100.)
self.assertEqual(gen3.cost_coeff_Q0, 0)
self.assertEqual(gen3.cost_coeff_Q1, 29.1 * 100)
self.assertEqual(gen3.cost_coeff_Q2, 0.00006 * (100**2.))
self.assertEqual(gen4.P, 20. / 100.)
self.assertEqual(gen4.P_min, 0)
self.assertEqual(gen4.P_max, 800. / 100.)
self.assertEqual(gen4.cost_coeff_Q0, 0)
self.assertEqual(gen4.cost_coeff_Q1, 6.9 * 100)
self.assertEqual(gen4.cost_coeff_Q2, 0.00026 * (100**2.))
self.assertEqual(gen5.P, 10. / 100.)
self.assertEqual(gen5.P_min, 0)
self.assertEqual(gen5.P_max, 2000. / 100.)
self.assertEqual(gen5.cost_coeff_Q0, 0)
self.assertEqual(gen5.cost_coeff_Q1, 50. * 100)
self.assertEqual(gen5.cost_coeff_Q2, 0.0015 * (100**2.))
# branches
branch1 = net.get_branch(12)
branch2 = net.get_branch(11)
branch3 = net.get_branch(10)
branch4 = net.get_branch(9)
branch5 = net.get_branch(8)
branch6 = net.get_branch(7)
branch7 = net.get_branch(6)
branch8 = net.get_branch(5)
branch9 = net.get_branch(4)
branch10 = net.get_branch(3)
branch11 = net.get_branch(2)
branch12 = net.get_branch(1)
branch13 = net.get_branch(0)
self.assertEqual(branch1.bus_k, bus1)
self.assertEqual(branch1.bus_m, bus3)
self.assertLess(abs(branch1.b + 1. / 0.1), 1e-10)
self.assertEqual(branch1.ratingA, 3000. / 100.)
self.assertEqual(branch1.ratingB, 3000. / 100.)
self.assertEqual(branch1.ratingC, 3000. / 100.)
self.assertEqual(branch2.bus_k, bus1)
self.assertEqual(branch2.bus_m, bus10)
self.assertLess(abs(branch2.b + 1. / 0.27), 1e-10)
self.assertEqual(branch2.ratingA, 2000. / 100.)
self.assertEqual(branch2.ratingB, 2000. / 100.)
self.assertEqual(branch2.ratingC, 2000. / 100.)
self.assertEqual(branch3.bus_k, bus2)
self.assertEqual(branch3.bus_m, bus3)
self.assertLess(abs(branch3.b + 1. / 0.12), 1e-10)
self.assertEqual(branch3.ratingA, 6500. / 100.)
self.assertEqual(branch3.ratingB, 6500. / 100.)
self.assertEqual(branch3.ratingC, 6500. / 100.)
self.assertEqual(branch4.bus_k, bus2)
self.assertEqual(branch4.bus_m, bus9)
self.assertLess(abs(branch4.b + 1. / 0.07), 1e-10)
self.assertEqual(branch4.ratingA, 5500. / 100.)
self.assertEqual(branch4.ratingB, 5500. / 100.)
self.assertEqual(branch4.ratingC, 5500. / 100.)
self.assertEqual(branch5.bus_k, bus2)
self.assertEqual(branch5.bus_m, bus10)
self.assertLess(abs(branch5.b + 1. / 0.14), 1e-10)
self.assertEqual(branch5.ratingA, 5500. / 100.)
self.assertEqual(branch5.ratingB, 5500. / 100.)
self.assertEqual(branch5.ratingC, 5500. / 100.)
self.assertEqual(branch6.bus_k, bus3)
self.assertEqual(branch6.bus_m, bus4)
self.assertLess(abs(branch6.b + 1. / 0.1), 1e-10)
self.assertEqual(branch6.ratingA, 3000. / 100.)
self.assertEqual(branch6.ratingB, 3000. / 100.)
self.assertEqual(branch6.ratingC, 3000. / 100.)
self.assertEqual(branch7.bus_k, bus3)
self.assertEqual(branch7.bus_m, bus5)
self.assertLess(abs(branch7.b + 1. / 0.17), 1e-10)
self.assertEqual(branch7.ratingA, 4000. / 100.)
self.assertEqual(branch7.ratingB, 4000. / 100.)
self.assertEqual(branch7.ratingC, 4000. / 100.)
self.assertEqual(branch8.bus_k, bus4)
self.assertEqual(branch8.bus_m, bus5)
self.assertLess(abs(branch8.b + 1. / 0.17), 1e-10)
self.assertEqual(branch8.ratingA, 4000. / 100.)
self.assertEqual(branch8.ratingB, 4000. / 100.)
self.assertEqual(branch8.ratingC, 4000. / 100.)
self.assertEqual(branch9.bus_k, bus5)
self.assertEqual(branch9.bus_m, bus6)
self.assertLess(abs(branch9.b + 1. / 0.17), 1e-10)
self.assertEqual(branch9.ratingA, 5000. / 100.)
self.assertEqual(branch9.ratingB, 5000. / 100.)
self.assertEqual(branch9.ratingC, 5000. / 100.)
self.assertEqual(branch10.bus_k, bus6)
self.assertEqual(branch10.bus_m, bus7)
self.assertLess(abs(branch10.b + 1. / 0.16), 1e-10)
self.assertEqual(branch10.ratingA, 2000. / 100.)
self.assertEqual(branch10.ratingB, 2000. / 100.)
self.assertEqual(branch10.ratingC, 2000. / 100.)
self.assertEqual(branch11.bus_k, bus7)
self.assertEqual(branch11.bus_m, bus8)
self.assertLess(abs(branch11.b + 1. / 0.25), 1e-10)
self.assertEqual(branch11.ratingA, 3000. / 100.)
self.assertEqual(branch11.ratingB, 3000. / 100.)
self.assertEqual(branch11.ratingC, 3000. / 100.)
self.assertEqual(branch12.bus_k, bus8)
self.assertEqual(branch12.bus_m, bus9)
self.assertLess(abs(branch12.b + 1. / 0.25), 1e-10)
self.assertEqual(branch12.ratingA, 2500. / 100.)
self.assertEqual(branch12.ratingB, 2500. / 100.)
self.assertEqual(branch12.ratingC, 2500. / 100.)
self.assertEqual(branch13.bus_k, bus8)
self.assertEqual(branch13.bus_m, bus10)
self.assertLess(abs(branch13.b + 1. / 0.07), 1e-10)
self.assertEqual(branch13.ratingA, 4000. / 100.)
self.assertEqual(branch13.ratingB, 4000. / 100.)
self.assertEqual(branch13.ratingC, 4000. / 100.)
def test_cas32art(self):
for case in test_cases.CASES:
if case == './data/case32.art':
net = pf.ParserART().parse(case)
self.assertEqual(net.num_buses, 31)
self.assertEqual(net.num_batteries, 3)
b1 = net.get_battery(0)
b2 = net.get_battery(1)
b3 = net.get_battery(2)
self.assertRaises(pf.NetworkError, net.get_battery, 3)
self.assertEqual(b1.bus.name, "N18")
self.assertEqual(b1.index, 0)
self.assertTrue(
all(
list(
map(lambda y: isinstance(y, pf.Battery),
b1.bus.batteries))))
self.assertEqual(len(b1.bus.batteries), 2)
self.assertEqual(
list(map(lambda y: y.index, b1.bus.batteries)),
[b1.index, b2.index])
self.assertEqual(b1.P, 6. / net.base_power)
self.assertEqual(b1.P_min, -7. / net.base_power)
self.assertEqual(b1.P_max, 8. / net.base_power)
self.assertEqual(b1.E, 14. / net.base_power)
self.assertEqual(b1.E_max, 22. / net.base_power)
self.assertEqual(b1.eta_c, 0.93)
self.assertEqual(b1.eta_d, 0.97)
self.assertEqual(b2.bus.name, "N18")
self.assertEqual(b2.index, 1)
self.assertTrue(
all(
list(
map(lambda y: isinstance(y, pf.Battery),
b2.bus.batteries))))
self.assertEqual(len(b2.bus.batteries), 2)
self.assertEqual(
list(map(lambda y: y.index, b2.bus.batteries)),
[b1.index, b2.index])
self.assertEqual(b2.P, 3. / net.base_power)
self.assertEqual(b2.P_min, -3. / net.base_power)
self.assertEqual(b2.P_max, 9. / net.base_power)
self.assertEqual(b2.E, 12. / net.base_power)
self.assertEqual(b2.E_max, 21. / net.base_power)
self.assertEqual(b2.eta_c, 0.94)
self.assertEqual(b2.eta_d, 0.92)
self.assertEqual(b3.bus.name, "N15")
self.assertEqual(b3.index, 2)
self.assertTrue(
all(
list(
map(lambda y: isinstance(y, pf.Battery),
b3.bus.batteries))))
self.assertEqual(len(b3.bus.batteries), 1)
self.assertEqual(
list(map(lambda y: y.index, b3.bus.batteries)), [b3.index])
self.assertEqual(b3.P, 2. / net.base_power)
self.assertEqual(b3.P_min, -4. / net.base_power)
self.assertEqual(b3.P_max, 5. / net.base_power)
self.assertEqual(b3.E, 10. / net.base_power)
self.assertEqual(b3.E_max, 20. / net.base_power)
self.assertEqual(b3.eta_c, 0.95)
self.assertEqual(b3.eta_d, 0.93)
def test_ieee14_gen_cost(self):
for case in test_cases.CASES:
if case == './data/ieee14.mat':
net = pf.ParserMAT().parse(case)
self.assertEqual(net.base_power, 100.)
self.assertEqual(net.num_buses, 14)
self.assertEqual(net.num_generators, 5)
gen0 = net.get_generator(net.num_generators - 1)
gen1 = net.get_generator(net.num_generators - 2)
self.assertEqual(gen0.P, 232.4 / 100.)
self.assertEqual(gen0.cost_coeff_Q2,
(4.3029259900e-02) * (net.base_power**2.))
self.assertEqual(gen0.cost_coeff_Q1, 20. * net.base_power)
self.assertEqual(gen1.P, 40. / 100.)
self.assertEqual(gen1.cost_coeff_Q2,
0.25 * (net.base_power**2.))
self.assertEqual(gen1.cost_coeff_Q1, 20. * net.base_power)
def test_type_parsers(self):
for case in test_cases.CASES:
if case.split('.')[-1] == 'mat':
self.assertRaises(pf.ParserError, pf.ParserART().parse, case)
self.assertRaises(pf.ParserError, pf.ParserJSON().parse, case)
if pf.info['raw_parser']:
self.assertRaises(pf.ParserError,
pf.ParserRAW().parse, case)
net = pf.ParserMAT().parse(case)
self.assertGreater(net.num_buses, 0)
elif case.split('.')[-1] == 'art':
self.assertRaises(pf.ParserError, pf.ParserMAT().parse, case)
self.assertRaises(pf.ParserError, pf.ParserJSON().parse, case)
if pf.info['raw_parser']:
self.assertRaises(pf.ParserError,
pf.ParserRAW().parse, case)
net = pf.ParserART().parse(case)
self.assertGreater(net.num_buses, 0)
elif case.split('.')[-1] == 'raw':
self.assertRaises(pf.ParserError, pf.ParserMAT().parse, case)
self.assertRaises(pf.ParserError, pf.ParserART().parse, case)
self.assertRaises(pf.ParserError, pf.ParserJSON().parse, case)
if pf.info['raw_parser']:
net = pf.ParserRAW().parse(case)
self.assertGreater(net.num_buses, 0)