def test_p2p(graphscope_session, p2p_31_e, p2p_31_v):
graph = graphscope.Graph(graphscope_session)
graph = graph.add_vertices(Loader(p2p_31_v, session=graphscope_session),
"person")
graph = graph.add_edges(Loader(p2p_31_e, session=graphscope_session),
"knows")
assert graph.schema is not None
def test_group(graphscope_session, student_group_e, student_v):
graph = graphscope.Graph(graphscope_session)
graph = graph.add_vertices(Loader(student_v, session=graphscope_session),
"student")
graph = graph.add_edges(
Loader(student_group_e, session=graphscope_session), "group")
assert graph.schema is not None
def ldbc_sample_multi_labels(prefix, directed):
graph = graphscope.Graph(directed=directed)
graph = (
graph.add_vertices(
Loader(os.path.join(prefix, "comment_0_0.csv"), delimiter="|"), "comment"
)
.add_vertices(
Loader(os.path.join(prefix, "person_0_0.csv"), delimiter="|"), "person"
)
.add_vertices(
Loader(os.path.join(prefix, "post_0_0.csv"), delimiter="|"),
"post",
)
)
graph = graph.add_edges(
Loader(os.path.join(prefix, "comment_replyOf_comment_0_0.csv"), delimiter="|"),
"replyOf",
src_label="comment",
dst_label="comment",
).add_edges(
Loader(os.path.join(prefix, "person_knows_person_0_0.csv"), delimiter="|"),
"knows",
["creationDate"],
src_label="person",
dst_label="person",
)
return graph
def p2p_property_graph(num_workers, directed=True):
data_dir = os.path.expandvars("${GS_TEST_DIR}/property")
graphscope.set_option(show_log=True)
graphscope.set_option(initializing_interactive_engine=False)
sess = graphscope.session(num_workers=num_workers, run_on_local=True)
graph = graphscope.Graph(sess, directed=directed)
graph = graph.add_vertices("{}/p2p-31_property_v_0".format(data_dir), "person")
graph = graph.add_edges("{}/p2p-31_property_e_0".format(data_dir), "knows")
return sess, graph
def ldbc_sample_single_label(prefix, directed):
graph = graphscope.Graph(directed=directed)
graph = graph.add_vertices(
Loader(os.path.join(prefix, "comment_0_0.csv"), delimiter="|"), "comment"
)
graph = graph.add_edges(
Loader(os.path.join(prefix, "comment_replyOf_comment_0_0.csv"), delimiter="|"),
"replyOf",
)
return graph
def load_subgraph(name):
import vineyard
import graphscope
graph = graphscope.Graph(self._session, generate_eid=False)
graph = graph.add_vertices(
Loader(vineyard.ObjectName("__%s_vertex_stream" % name)))
graph = graph.add_edges(
Loader(vineyard.ObjectName("__%s_edge_stream" % name)))
graph._ensure_loaded()
logger.info("subgraph has been loaded")
return graph
def ldbc_sample_with_duplicated_oid(prefix, directed):
graph = graphscope.Graph(directed=directed)
graph = graph.add_vertices(
Loader(os.path.join(prefix, "place_0_0.csv"), delimiter="|"), "place"
).add_vertices(
Loader(os.path.join(prefix, "person_0_0.csv"), delimiter="|"), "person"
)
graph = graph.add_edges(
Loader(os.path.join(prefix, "place_isPartOf_place_0_0.csv"), delimiter="|"),
"isPartOf",
src_label="place",
dst_label="place",
).add_edges(
Loader(os.path.join(prefix, "person_knows_person_0_0.csv"), delimiter="|"),
"knows",
["creationDate"],
src_label="person",
dst_label="person",
)
return graph