def test_Vertex_add_directed_edge_2(): vertex = Vertex(1) edge_1 = Vertex(3) edge_2 = Vertex(4) vertex.add_directed_edge(edge_1) vertex.add_directed_edge(edge_2) assert vertex.directed_edges == {edge_1, edge_2}
def test_Vertex_add_directed_edge_1(): vertex = Vertex(1) edge_1 = Vertex(2) vertex.add_directed_edge(edge_1) assert vertex.directed_edges == {edge_1}