def test_remove_outlet(self, outlet1, outlet2): link = Link([], [], timedelta(seconds=1), tags='test_remove_outlet') link.add_outlets([outlet1, outlet2]) link.remove_outlets(outlet2) self.assertEqual(link.outlets, [outlet1])
def test_add_outlet_same(self, outlet1): link = Link([], [], timedelta(seconds=1), tags='test_add_outlet_same') link.add_outlets(outlet1) self.assertRaises(InvalidNodeError, link.add_outlets, outlet1) self.assertEqual(link.outlets, [outlet1])
def test_add_outlet_multiple(self, outlet1, outlet2): link = Link([], [], timedelta(seconds=1), tags='test_add_outlet_multiple') link.add_outlets([outlet1, outlet2]) self.assertEqual(link.outlets, [outlet1, outlet2])
def test_remove_outlet_invalid(self, outlet1, outlet2): link = Link([], [], timedelta(seconds=1), tags='test_remove_outlet_invalid') link.add_outlets([outlet1]) self.assertRaises(InvalidNodeError, link.remove_outlets, outlet2) self.assertEqual(link.outlets, [outlet1])
def test_add_outlet(self, outlet1): link = Link([], [], timedelta(seconds=1), tags='test_add_outlet') link.add_outlets(outlet1) self.assertEqual(link.outlets, [outlet1])