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 == {}