Function test_add_column

python/graphscope/tests/unittest/test_graph.py:636–670

(ldbc_graph, arrow_modern_graph)

def test_add_column(ldbc_graph, arrow_modern_graph):
    ldbc = ldbc_graph
    modern = arrow_modern_graph

    sub_graph_1 = ldbc.project(vertices={"person": []}, edges={"knows": ["eid"]})
    sub_graph_2 = ldbc.project(
        vertices={"person": None, "tag": None}, edges={"hasInterest": None}
    )
    sub_graph_3 = ldbc.project(
        vertices={"tag": None, "tagclass": None}, edges={"hasType": None}
    )
    sub_graph_4 = modern.project(vertices={"person": []}, edges={"knows": ["eid"]})

    ret = graphscope.pagerank(sub_graph_1)

    # the ret can add to the graph queried on
    g1 = sub_graph_1.add_column(ret, selector={"pr": "r"})
    assert g1.schema.get_vertex_properties("person")[0].id == 8
    assert g1.schema.get_vertex_properties("person")[0].name == "pr"
    # the ret can add to the origin graph
    g2 = ldbc.add_column(ret, selector={"pr": "r"})
    assert g2.schema.get_vertex_properties("person")[8].id == 8
    assert g2.schema.get_vertex_properties("person")[8].name == "pr"
    # the ret can add to the graph tha contain the same vertex label with sub_graph_1
    g3 = sub_graph_2.add_column(ret, selector={"pr": "r"})
    assert g3.schema.get_vertex_properties("person")[8].id == 8
    assert g3.schema.get_vertex_properties("person")[8].name == "pr"
    # the ret cannot add to sub_graph_3
    with pytest.raises(AnalyticalEngineInternalError):
        g4 = sub_graph_3.add_column(ret, selector={"pr": "r"})
        print(g4.schema)
    # the ret cannot add to sub_graph_4
    with pytest.raises(AnalyticalEngineInternalError):
        g5 = sub_graph_4.add_column(ret, selector={"pr": "r"})
        print(g4.schema)


def test_add_column_string_oid(