Method PEval

python/graphscope/tests/unittest/test_udf_app.py:137–218

(frag, context)

    @staticmethod
    def PEval(frag, context):
        graphscope.declare(graphscope.Vertex, node)
        fid = frag.fid()
        if fid == 0:
            # This is not stable, as it depend on worker's number
            # assert frag.fnum() == 4
            assert frag.vertex_label_num() == 8
            assert frag.edge_label_num() == 15
            assert frag.get_total_nodes_num() == 190376

            v_label_num = frag.vertex_label_num()
            for v_label_id in range(v_label_num):
                assert frag.get_nodes_num(v_label_id) == (
                    frag.get_inner_nodes_num(v_label_id)
                    + frag.get_outer_nodes_num(v_label_id)
                )

                nodes = frag.nodes(v_label_id)
                assert nodes.size() == frag.get_nodes_num(v_label_id)

                inner_nodes = frag.inner_nodes(v_label_id)
                assert inner_nodes.size() == frag.get_inner_nodes_num(v_label_id)

                outer_nodes = frag.outer_nodes(v_label_id)
                assert outer_nodes.size() == frag.get_outer_nodes_num(v_label_id)

                for iv in inner_nodes:
                    assert frag.get_node_fid(iv) == 0
                    assert frag.is_inner_node(iv) == True
                    vid = frag.get_node_id(iv)
                    assert frag.get_node(v_label_id, vid, node) == True
                    assert frag.get_inner_node(v_label_id, vid, node) == True

                    e_label_num = frag.edge_label_num()
                    for e_label_id in range(e_label_num):
                        edges = frag.get_outgoing_edges(iv, e_label_id)
                        if edges.size() > 0:
                            assert frag.has_child(iv, e_label_id) == True
                        assert frag.get_outdegree(iv, e_label_id) == int(edges.size())

                        edges = frag.get_incoming_edges(iv, e_label_id)
                        if edges.size() > 0:
                            assert frag.has_parent(iv, e_label_id) == True
                        assert frag.get_indegree(iv, e_label_id) == int(edges.size())

                for ov in outer_nodes:
                    assert frag.is_outer_node(ov) == True
                    vid = frag.get_node_id(ov)
                    assert frag.get_node(v_label_id, vid, node) == True
                    assert frag.get_outer_node(v_label_id, vid, node) == True

            for v_label in frag.vertex_labels():
                label_id = frag.get_vertex_label_id_by_name(v_label)
                assert frag.get_vertex_label_by_id(label_id) == v_label

                for prop in frag.vertex_properties(v_label):
                    prop_id = frag.get_vertex_property_id_by_name(v_label, prop.first)
                    assert (