Make all benchmarks cases similar so is easier to compare

Author: syrusakbary

graphql/execution/querybuilder/tests/test_querybuilder.py

@@ -106,6 +106,7 @@ def test_fragment_operation_mutation():
         execute_serially=True
     )
 
+
 def test_query_builder_operation():
     Node = GraphQLObjectType('Node', fields={'id': GraphQLField(GraphQLInt)})
     Query = GraphQLObjectType('Query', fields={'nodes': GraphQLField(GraphQLList(Node))})
@@ -144,3 +145,24 @@ def test_query_builder_operation():
             )
         }
     )
+
+
+def test_query_builder_execution():
+    Node = GraphQLObjectType('Node', fields={'id': GraphQLField(GraphQLInt, resolver=lambda obj, **__: obj)})
+    Query = GraphQLObjectType('Query', fields={'nodes': GraphQLField(GraphQLList(Node), resolver=lambda *_, **__: range(3))})
+
+    schema = GraphQLSchema(query=Query)
+    document_ast = parse('''query MyQuery {
+        nodes {
+            id
+        }
+    }''')
+    query_builder = QueryBuilder(schema, document_ast)
+    QueryFragment = query_builder.get_operation_fragment('MyQuery')
+    root = None
+    expected = {
+        'nodes': [{
+            'id': n
+        } for n in range(3)]
+    }
+    assert QueryFragment.resolver(lambda: root) == expected

graphql/execution/tests/test_benchmark.py

@@ -5,8 +5,10 @@
                      GraphQLSchema, Source, execute, parse)
 
 
+SIZE = 10000
+
 def test_big_list_of_ints(benchmark):
-    big_int_list = [x for x in range(5000)]
+    big_int_list = [x for x in range(SIZE)]
 
     def resolve_all_ints(root, args, context, info):
         return big_int_list
@@ -28,8 +30,8 @@ def resolve_all_ints(root, args, context, info):
 
 
 
-def test_big_list_of_ints_base(benchmark):
-    big_int_list = [x for x in range(5000)]
+def test_big_list_of_ints_only_serialize(benchmark):
+    big_int_list = [x for x in range(SIZE)]
     from ..executor import complete_leaf_value
     # def convert_item(i):
     #     return i
@@ -42,6 +44,32 @@ def convert_list():
     benchmark(convert_list)
 
 
+def test_big_list_of_objecttypes_with_one_int_field(benchmark):
+    ContainerType = GraphQLObjectType('Container', fields={
+        'x': GraphQLField(GraphQLInt, resolver=lambda root, args, context, info: root),
+    })
+
+    big_container_list = [x for x in range(SIZE)]
+
+    def resolve_all_containers(root, args, context, info):
+        return big_container_list
+
+    Query = GraphQLObjectType('Query', fields={
+        'allContainers': GraphQLField(
+            GraphQLList(ContainerType),
+            resolver=resolve_all_containers
+        )
+    })
+    hello_schema = GraphQLSchema(Query)
+    source = Source('{ allContainers { x } }')
+    ast = parse(source)
+    big_list_query = partial(execute, hello_schema, ast)
+    result = benchmark(big_list_query)
+    # result = big_list_query()
+    assert not result.errors
+    assert result.data == {'allContainers': [{'x': x} for x in big_container_list]}
+
+
 def test_big_list_of_containers_with_one_field(benchmark):
     Container = namedtuple('Container', 'x y z o')
 
@@ -52,7 +80,7 @@ def test_big_list_of_containers_with_one_field(benchmark):
         'o': GraphQLField(GraphQLInt),
     })
 
-    big_container_list = [Container(x=x, y=x, z=x, o=x) for x in range(5000)]
+    big_container_list = [Container(x=x, y=x, z=x, o=x) for x in range(SIZE)]
 
     def resolve_all_containers(root, args, context, info):
         return big_container_list
@@ -83,7 +111,7 @@ def test_big_list_of_containers_with_multiple_fields(benchmark):
         'o': GraphQLField(GraphQLInt),
     })
 
-    big_container_list = [Container(x=x, y=x, z=x, o=x) for x in range(5000)]
+    big_container_list = [Container(x=x, y=x, z=x, o=x) for x in range(SIZE)]
 
     def resolve_all_containers(root, args, context, info):
         return big_container_list