def test_setNodeB(self): """Test le fonctionnement de la fonction setNodeB()""" nodeA = Node(1, 5, 10) nodeB = Node(2, 6, 8) edge = Edge(nodeA, nodeB) nodeC = Node(3, 7, 20) edge.setNodeB(nodeC) #test changement nodeB self.assertIs(nodeC, edge.getNodeB())
def test_init(self): """Test du __init__""" nodeA = Node(1, 5, 10) nodeB = Node(2, 6, 8) edge = Edge(nodeA, nodeB) self.assertIs(nodeA, edge.getNodeA()) self.assertIs(nodeB, edge.getNodeB()) self.assertFalse(edge.isDirected()) self.assertFalse(edge.isDirectedA()) self.assertFalse(edge.isDirectedB()) self.assertEqual({}, edge.getFeatures())