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