def test_remove_inlet(self, inlet1, inlet2):
link = Link([], [], timedelta(seconds=1), tags='test_remove_inlet')
link.add_inlets([inlet1, inlet2])
link.remove_inlets(inlet2)
self.assertEqual(link.inlets, [inlet1])
def test_add_inlet_same(self, inlet1):
link = Link([], [], timedelta(seconds=1), tags='test_add_inlet_same')
link.add_inlets(inlet1)
self.assertRaises(InvalidNodeError, link.add_inlets, inlet1)
self.assertEqual(link.inlets, [inlet1])
def test_add_inlet_multiple(self, inlet1, inlet2):
link = Link([], [],
timedelta(seconds=1),
tags='test_add_inlet_multiple')
link.add_inlets([inlet1, inlet2])
self.assertEqual(link.inlets, [inlet1, inlet2])
def test_remove_inlet_invalid(self, inlet1, inlet2):
link = Link([], [],
timedelta(seconds=1),
tags='test_remove_inlet_invalid')
link.add_inlets([inlet1])
self.assertRaises(InvalidNodeError, link.remove_inlets, inlet2)
self.assertEqual(link.inlets, [inlet1])
def test_add_multiple_inlets_same(self):
inlet1 = DummyInlet()
inlet2 = DummyInlet()
link = Link([], [], timedelta(seconds=1), tags='test_add_inlet_same')
link.add_inlets(inlet1)
link.add_inlets(inlet2)
self.assertEqual(link.inlets, [inlet1, inlet2])
def test_add_inlet(self, inlet1):
link = Link([], [], timedelta(seconds=1), tags='test_add_inlet')
link.add_inlets(inlet1)
self.assertEqual(link.inlets, [inlet1])
