def test_addFeature(self): """Test le fonctionnement de la fonctions addFeature""" nodeA = Node(1, 5, 10) nodeB = Node(2, 6, 8) edge = Edge(nodeA, nodeB) #test features vide self.assertEqual({}, edge.getFeatures()) edge.addFeature('test','value') #test ajout d'une feature self.assertIn('test', edge.getFeatures()) #test contenue feature 'test self.assertEqual('value', edge.getFeature('test'))
def test_removeAllFeature(self): """Test le fonctionnement de la fonctions removeFeature""" nodeA = Node(1, 5, 10) nodeB = Node(2, 6, 8) edge = Edge(nodeA, nodeB) #test features vide self.assertEqual({}, edge.getFeatures()) edge.addFeature('test1','value') #test ajout d'une feature self.assertIn('test1', edge.getFeatures()) #test contenue feature 'test self.assertEqual('value', edge.getFeature('test1')) edge.addFeature('test2','value2') #test ajout nouvelle feature self.assertIn('test1', edge.getFeatures()) self.assertIn('test2', edge.getFeatures()) #test removeAll edge.removeAllFeatures() self.assertEqual({}, edge.getFeatures())
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())