def test_get_connected_buses():
net = pp.create_empty_network()
bus0 = pp.create_bus(net, vn_kv=110)
bus1 = pp.create_bus(net, vn_kv=20)
bus2 = pp.create_bus(net, vn_kv=10)
bus3 = pp.create_bus(net, vn_kv=0.4)
bus4 = pp.create_bus(net, vn_kv=0.4)
bus5 = pp.create_bus(net, vn_kv=20)
trafo0 = pp.create_transformer3w(net, hv_bus=bus0, mv_bus=bus1, lv_bus=bus2, name='trafo0',
std_type='63/25/38 MVA 110/20/10 kV')
trafo1 = pp.create_transformer(net, hv_bus=bus2, lv_bus=bus3, std_type='0.4 MVA 10/0.4 kV')
line1 = pp.create_line(net, from_bus=bus3, to_bus=bus4, length_km=20.1, std_type='24-AL1/4-ST1A 0.4', name='line1')
# switch0=pp.create_switch(net, bus = bus0, element = trafo0, et = 't3') #~~~~~ not implementable
switch1 = pp.create_switch(net, bus=bus1, element=bus5, et='b')
switch2 = pp.create_switch(net, bus=bus2, element=trafo1, et='t')
switch3 = pp.create_switch(net, bus=bus3, element=line1, et='l')
assert list(tb.get_connected_buses(net, [bus0])) == [bus1, bus2] # trafo3w has not been implemented in the function
assert list(tb.get_connected_buses(net, [bus1])) == [bus0, bus2,
bus5] # trafo3w has not been implemented in the function
assert list(tb.get_connected_buses(net, [bus2])) == [bus0, bus1,
bus3] # trafo3w has not been implemented in the function
assert list(tb.get_connected_buses(net, [bus3])) == [bus2, bus4]
assert list(tb.get_connected_buses(net, [bus4])) == [bus3]
assert list(tb.get_connected_buses(net, [bus5])) == [bus1]
assert list(tb.get_connected_buses(net, [bus0, bus1])) == [bus2, bus5]
assert list(tb.get_connected_buses(net, [bus2, bus3])) == [bus0, bus1, bus4]
def test_get_connected_buses():
net = pp.create_empty_network()
bus0 = pp.create_bus(net, vn_kv=110)
bus1 = pp.create_bus(net, vn_kv=20)
bus2 = pp.create_bus(net, vn_kv=10)
bus3 = pp.create_bus(net, vn_kv=0.4)
bus4 = pp.create_bus(net, vn_kv=0.4)
bus5 = pp.create_bus(net, vn_kv=20)
trafo0 = pp.create_transformer3w(net, hv_bus=bus0, mv_bus=bus1, lv_bus=bus2, std_type='63/25/38 MVA 110/20/10 kV')
trafo1 = pp.create_transformer(net, hv_bus=bus2, lv_bus=bus3, std_type='0.4 MVA 10/0.4 kV')
line1 = pp.create_line(net, from_bus=bus3, to_bus=bus4, length_km=20.1, std_type='24-AL1/4-ST1A 0.4')
switch0a = pp.create_switch(net, bus=bus0, element=trafo0, et='t3')
switch0b = pp.create_switch(net, bus=bus2, element=trafo0, et='t3')
switch1 = pp.create_switch(net, bus=bus1, element=bus5, et='b')
switch2 = pp.create_switch(net, bus=bus2, element=trafo1, et='t')
switch3 = pp.create_switch(net, bus=bus3, element=line1, et='l')
assert list(tb.get_connected_buses(net, [bus0])) == [bus1, bus2]
assert list(tb.get_connected_buses(net, [bus1])) == [bus0, bus2, bus5]
assert list(tb.get_connected_buses(net, [bus2])) == [bus0, bus1, bus3]
assert list(tb.get_connected_buses(net, [bus3])) == [bus2, bus4]
assert list(tb.get_connected_buses(net, [bus4])) == [bus3]
assert list(tb.get_connected_buses(net, [bus5])) == [bus1]
assert list(tb.get_connected_buses(net, [bus0, bus1])) == [bus2, bus5]
assert list(tb.get_connected_buses(net, [bus2, bus3])) == [bus0, bus1, bus4]
net.switch.loc[[switch0b, switch1, switch2, switch3], 'closed'] = False
assert list(tb.get_connected_buses(net, [bus0])) == [bus1]
assert list(tb.get_connected_buses(net, [bus1])) == [bus0]
assert list(tb.get_connected_buses(net, [bus3])) == []
assert list(tb.get_connected_buses(net, [bus4])) == []