From c6ed12df833d676ed05b471998a788ef2561f681 Mon Sep 17 00:00:00 2001 From: siyuan0322 Date: Tue, 8 Jun 2021 22:40:31 +0800 Subject: [PATCH] add type --- python/tests/test_create_graph.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/python/tests/test_create_graph.py b/python/tests/test_create_graph.py index 849e55b7034d..40e102faf05e 100644 --- a/python/tests/test_create_graph.py +++ b/python/tests/test_create_graph.py @@ -27,6 +27,7 @@ from graphscope.framework.errors import AnalyticalEngineInternalError from graphscope.framework.graph import Graph from graphscope.framework.loader import Loader +from graphscope.proto import graph_def_pb2 @pytest.fixture @@ -245,12 +246,12 @@ def test_loader_with_specified_data_type( student_group_e, "group", ["group_id", ("member_size", "int")] ) assert [p.type for p in graph.schema.get_vertex_properties("student")] == [ - 10, - 4, - 8, - 10, + graph_def_pb2.STRING, + graph_def_pb2.INT, + graph_def_pb2.FLOAT, + graph_def_pb2.STRING ] - assert [p.type for p in graph.schema.get_edge_properties("group")] == [10, 4] + assert [p.type for p in graph.schema.get_edge_properties("group")] == [graph_def_pb2.STRING, graph_def_pb2.INT] def test_multi_src_dst_edge_loader(