def test_remove_neighbor(self):
vertex = Vertex(data='a')
self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
vertex.add_neighbor(data='b')
self.assertEqual(set(list(vertex.adjacent.keys())), set(['b']))
self.assertEqual(vertex.adjacent.get('b'), 1)
self.assertEqual(vertex.remove_neighbor(data='b'), True)
def test_remove_neighbor_key_error(self):
vertex = Vertex(data='a')
self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
vertex.add_neighbor(data='b')
self.assertEqual(set(list(vertex.adjacent.keys())), set(['b']))
self.assertEqual(vertex.adjacent.get('b'), 1)
with self.assertRaises(KeyError) as error:
self.assertEqual(vertex.remove_neighbor(data='invalid_key'), False)
