def test_read_csv_property_graph_from_string(): g = create_graph(directed=False, allowing_self_loops=False, allowing_multiple_edges=False, weighted=True, any_hashable=True, vertex_supplier=create_vertex_supplier(), edge_supplier=create_edge_supplier()) input_string = """1,2 2,3 3,4 4,1 """ def import_id_cb(id): return 'v{}'.format(int(id) + 1) parse_csv(g, input_string, import_id_cb=import_id_cb) print(g.vertices) assert g.vertices == {'v2', 'v3', 'v4', 'v5'} assert g.edge_tuple('e2') == ('v4', 'v5', 1.0) assert g.vertex_attrs == {} assert g.edge_attrs == {}
def test_input_csv_from_string_create_new_vertices(): g = create_graph(directed=False, allowing_self_loops=False, allowing_multiple_edges=False, weighted=True) input_string = """1,2 2,3 3,4 4,1 """ print(set(g.vertices())) parse_csv(g, input_string) print(set(g.vertices())) print(set(g.edges())) assert g.vertices() == set([0, 1, 2, 3])
def test_input_csv_from_string_preserve_ids(): g = create_graph(directed=False, allowing_self_loops=False, allowing_multiple_edges=False, weighted=True) input_string = """1,2 2,3 3,4 4,1 """ def import_id(file_id): return int(file_id) parse_csv(g, input_string, import_id_cb=import_id) print(set(g.vertices())) print(set(g.edges())) assert g.vertices() == set([1, 2, 3, 4])
def test_read_csv_property_graph_from_string1(): g = create_graph(directed=False, allowing_self_loops=False, allowing_multiple_edges=False, weighted=True, any_hashable=True, vertex_supplier=create_vertex_supplier(), edge_supplier=create_edge_supplier()) input_string = """1,2 2,3 3,4 4,1 """ parse_csv(g, input_string) print(g.vertices) assert g.vertices == {'v0', 'v1', 'v2', 'v3'} assert g.edge_tuple('e2') == ('v2', 'v3', 1.0) assert g.vertex_attrs == {} assert g.edge_attrs == {}