def test_bus_indexes(self):
""" Test the from/to bus index property.
"""
c = Case(name="c")
bus1 = Bus(name="Bus 1")
bus2 = Bus(name="Bus 2")
bus3 = Bus(name="Bus 3")
c.buses = [bus1, bus2, bus3]
# Append to list.
branch1 = Branch(bus3, bus1)
c.branches.append(branch1)
self.assertEqual(c.buses.index(branch1.from_bus), 2)
self.assertEqual(c.buses.index(branch1.to_bus), 0)
# Set list.
branch2 = Branch(bus2, bus3)
branch3 = Branch(bus2, bus1)
c.branches = [branch2, branch3]
self.assertEqual(c.buses.index(branch2.from_bus), 1)
self.assertEqual(c.buses.index(branch2.to_bus), 2)
# Move branch.
branch2.from_bus = bus1
self.assertEqual(c.buses.index(branch2.from_bus), 0)
def test_bus_indexes(self):
""" Test the from/to bus index property.
"""
c = Case(name="c")
bus1 = Bus(name="Bus 1")
bus2 = Bus(name="Bus 2")
bus3 = Bus(name="Bus 3")
c.buses = [bus1, bus2, bus3]
# Append to list.
branch1 = Branch(bus3, bus1)
c.branches.append(branch1)
self.assertEqual(c.buses.index(branch1.from_bus), 2)
self.assertEqual(c.buses.index(branch1.to_bus), 0)
# Set list.
branch2 = Branch(bus2, bus3)
branch3 = Branch(bus2, bus1)
c.branches = [branch2, branch3]
self.assertEqual(c.buses.index(branch2.from_bus), 1)
self.assertEqual(c.buses.index(branch2.to_bus), 2)
# Move branch.
branch2.from_bus = bus1
self.assertEqual(c.buses.index(branch2.from_bus), 0)
