def arrow_property_graph_lpa(graphscope_session): g = Graph(graphscope_session, generate_eid=False) g = g.add_vertices(f"{property_dir}/lpa_dataset/lpa_3000_v_0", "v0") g = g.add_vertices(f"{property_dir}/lpa_dataset/lpa_3000_v_1", "v1") g = g.add_edges(f"{property_dir}/lpa_dataset/lpa_3000_e_0", "e0", ["weight"], "v0", "v1") yield g g.unload()
def p2p_property_graph_undirected(graphscope_session): g = Graph(graphscope_session, directed=False, generate_eid=False) g = g.add_vertices(f"{property_dir}/p2p-31_property_v_0", "person") g = g.add_edges( f"{property_dir}/p2p-31_property_e_0", label="knows", src_label="person", dst_label="person", ) yield g g.unload()
def test_load_graph_copy(graphscope_session, arrow_property_graph): g = arrow_property_graph g2 = Graph(graphscope_session, g) assert g.key != g2.key assert g.vineyard_id != g2.vineyard_id assert str(g.schema) == str(g2.schema) assert np.all(g.to_numpy("v:v0.id") == g2.to_numpy("v:v0.id")) g2.unload() assert not g2.loaded() # test load from vineyard's graph g3 = Graph(graphscope_session, vineyard.ObjectID(g.vineyard_id)) assert g3.loaded()
def test_error_on_operation_on_graph(graphscope_session): g = Graph(graphscope_session.session_id) with pytest.raises(RuntimeError, match="The graph is not registered in remote"): g.project_to_simple(v_label=0, v_prop=0, e_label=0, e_prop=0) with pytest.raises(RuntimeError, match="The graph is not registered in remote"): g.unload() with pytest.raises(AssertionError): property_sssp(g, src=6)
def arrow_property_graph_only_from_efile(graphscope_session): g = Graph(graphscope_session, generate_eid=False) g = g.add_edges(f"{new_property_dir}/twitter_e_0_0_0", "e0", ["weight"], "v0", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_0_1_0", "e0", ["weight"], "v0", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_1_0_0", "e0", ["weight"], "v1", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_1_1_0", "e0", ["weight"], "v1", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_0_0_1", "e1", ["weight"], "v0", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_0_1_1", "e1", ["weight"], "v0", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_1_0_1", "e1", ["weight"], "v1", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_1_1_1", "e1", ["weight"], "v1", "v1") yield g g.unload()
def test_unload(graphscope_session): graph = Graph(graphscope_session) prefix = os.path.expandvars("${GS_TEST_DIR}/property") graph = (Graph(graphscope_session).add_vertices( f"{prefix}/p2p-31_property_v_0", "person").add_edges(f"{prefix}/p2p-31_property_e_0", "knows")) assert graph.loaded() assert graph.vineyard_id is not None graph.unload() assert not graph.loaded() with pytest.raises(RuntimeError, match="The graph is not loaded"): graph.unload() with pytest.raises(RuntimeError, match="The graph is not loaded"): graph.project_to_simple(v_label="person", e_label="knows") with pytest.raises(AssertionError): g2 = Graph(graphscope_session, graph) with pytest.raises(RuntimeError, match="The graph is not loaded"): property_sssp(graph, src=6)
def arrow_property_graph_undirected(graphscope_session): g = Graph(graphscope_session, directed=False, generate_eid=False) g = g.add_vertices(f"{new_property_dir}/twitter_v_0", "v0") g = g.add_vertices(f"{new_property_dir}/twitter_v_1", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_0_0_0", "e0", ["weight"], "v0", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_0_1_0", "e0", ["weight"], "v0", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_1_0_0", "e0", ["weight"], "v1", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_1_1_0", "e0", ["weight"], "v1", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_0_0_1", "e1", ["weight"], "v0", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_0_1_1", "e1", ["weight"], "v0", "v1") g = g.add_edges(f"{new_property_dir}/twitter_e_1_0_1", "e1", ["weight"], "v1", "v0") g = g.add_edges(f"{new_property_dir}/twitter_e_1_1_1", "e1", ["weight"], "v1", "v1") yield g g.unload()