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)