Fixed benchmark tests

This commit is contained in:
Syrus Akbary 2017-07-31 22:15:26 -07:00
parent 602d8866bb
commit 7f1c2a654b

View File

@ -340,7 +340,7 @@ def test_big_list_of_containers_query_benchmark(benchmark):
class Query(ObjectType): class Query(ObjectType):
all_containers = List(Container) all_containers = List(Container)
def resolve_all_containers(self): def resolve_all_containers(self, info):
return big_container_list return big_container_list
hello_schema = Schema(Query) hello_schema = Schema(Query)
@ -363,7 +363,7 @@ def test_big_list_of_containers_multiple_fields_query_benchmark(benchmark):
class Query(ObjectType): class Query(ObjectType):
all_containers = List(Container) all_containers = List(Container)
def resolve_all_containers(self): def resolve_all_containers(self, info):
return big_container_list return big_container_list
hello_schema = Schema(Query) hello_schema = Schema(Query)
@ -381,16 +381,16 @@ def test_big_list_of_containers_multiple_fields_custom_resolvers_query_benchmark
z = Int() z = Int()
o = Int() o = Int()
def resolve_x(self): def resolve_x(self, info):
return self.x return self.x
def resolve_y(self): def resolve_y(self, info):
return self.y return self.y
def resolve_z(self): def resolve_z(self, info):
return self.z return self.z
def resolve_o(self): def resolve_o(self, info):
return self.o return self.o
big_container_list = [Container(x=x, y=x, z=x, o=x) for x in range(1000)] big_container_list = [Container(x=x, y=x, z=x, o=x) for x in range(1000)]
@ -398,7 +398,7 @@ def test_big_list_of_containers_multiple_fields_custom_resolvers_query_benchmark
class Query(ObjectType): class Query(ObjectType):
all_containers = List(Container) all_containers = List(Container)
def resolve_all_containers(self): def resolve_all_containers(self, info):
return big_container_list return big_container_list
hello_schema = Schema(Query) hello_schema = Schema(Query)