def test_get_weight(self):
vertex = Vertex(data='a')
self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
vertex.add_neighbor(data='b')
self.assertEqual(vertex.get_weight('b'), 1)
vertex.add_neighbor(data='b', weight=5)
self.assertEqual(vertex.get_weight('b'), 5)
def test_add(self):
vertex = Vertex(data='a')
self.assertEqual(vertex.add(neighbors=None), None)
self.assertEqual(set(list(vertex.get_neighbors())), set([]))
data = [
('b', 3),
('c', 7),
('d'),
tuple(['e'])
]
vertex.add(neighbors=data)
self.assertEqual(set(list(vertex.get_neighbors())),
set(['b', 'c', 'd', 'e']))
self.assertEqual(vertex.get_weight('b'), 3)
self.assertEqual(vertex.get_weight('c'), 7)
self.assertEqual(vertex.get_weight('d'), 1)
self.assertEqual(vertex.get_weight('e'), 1)
def test_get_weight_keyerror(self):
vertex = Vertex(data='a')
self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
vertex.add_neighbor(data='b')
with self.assertRaises(KeyError) as error:
self.assertEqual(vertex.get_weight('c'), None)
