def test_add_edges_correct_set_of_neighbors_test(self):
g = Graph()
v1, v2, v3 = self._get_vertex(0), self._get_vertex(1), self._get_vertex(2)
e_forward = Edge(v1.id, v2.id, True, False, EdgeData(1, " ", 100, "Test"))
e_backward = Edge(v2.id, v3.id, False, True, EdgeData(1, " ", 100, "Test"))
e_both = Edge(v3.id, v1.id, True, True, EdgeData(1, " ", 100, "Test"))
g.add_node(v1)
g.add_node(v2)
g.add_node(v3)
g.add_edge(e_forward)
g.add_edge(e_backward)
g.add_edge(e_both)
self.assertTrue(self._checkEqual([v2.id, v3.id], g.all_neighbors(v1.id)))
self.assertTrue(self._checkEqual([v1.id, v3.id], g.all_neighbors(v2.id)))
self.assertTrue(self._checkEqual([v1.id, v2.id], g.all_neighbors(v3.id)))
