Fix for type number changes. (#388)

alibaba · Jun 8, 2021 · 993d388 · 993d388
1 parent b8a2881
commit 993d388
Showing 1 changed file with 6 additions and 5 deletions.
diff --git 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(