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])