@@ -28,7 +28,6 @@ def test_batches_correctly(executor):
2828
2929 schema = GraphQLSchema (query = Query )
3030
31-
3231 doc = '''
3332{
3433 business1: getBusiness(id: "1") {
@@ -41,9 +40,8 @@ def test_batches_correctly(executor):
4140 '''
4241 doc_ast = parse (doc )
4342
44-
4543 load_calls = []
46-
44+
4745 class BusinessDataLoader (DataLoader ):
4846 def batch_load_fn (self , keys ):
4947 load_calls .append (keys )
@@ -52,7 +50,6 @@ def batch_load_fn(self, keys):
5250 class Context (object ):
5351 business_data_loader = BusinessDataLoader ()
5452
55-
5653 result = execute (schema , doc_ast , None , context_value = Context (), executor = executor )
5754 assert not result .errors
5855 assert result .data == {
@@ -63,7 +60,7 @@ class Context(object):
6360 'id' : '2'
6461 },
6562 }
66- assert load_calls == [['1' ,'2' ]]
63+ assert load_calls == [['1' , '2' ]]
6764
6865
6966@pytest .mark .parametrize ("executor" , [
@@ -78,8 +75,11 @@ def test_batches_multiple_together(executor):
7875
7976 Business = GraphQLObjectType ('Business' , lambda : {
8077 'id' : GraphQLField (GraphQLID , resolver = lambda root , info , ** args : root ),
81- 'location' : GraphQLField (Location ,
82- resolver = lambda root , info , ** args : info .context .location_data_loader .load ('location-{}' .format (root ))
78+ 'location' : GraphQLField (
79+ Location ,
80+ resolver = lambda root , info , ** args : info .context .location_data_loader .load (
81+ 'location-{}' .format (root )
82+ )
8383 ),
8484 })
8585
@@ -94,7 +94,6 @@ def test_batches_multiple_together(executor):
9494
9595 schema = GraphQLSchema (query = Query )
9696
97-
9897 doc = '''
9998{
10099 business1: getBusiness(id: "1") {
@@ -113,16 +112,15 @@ def test_batches_multiple_together(executor):
113112 '''
114113 doc_ast = parse (doc )
115114
116-
117115 business_load_calls = []
118-
116+
119117 class BusinessDataLoader (DataLoader ):
120118 def batch_load_fn (self , keys ):
121119 business_load_calls .append (keys )
122120 return Promise .resolve (keys )
123121
124122 location_load_calls = []
125-
123+
126124 class LocationDataLoader (DataLoader ):
127125 def batch_load_fn (self , keys ):
128126 location_load_calls .append (keys )
@@ -132,7 +130,6 @@ class Context(object):
132130 business_data_loader = BusinessDataLoader ()
133131 location_data_loader = LocationDataLoader ()
134132
135-
136133 result = execute (schema , doc_ast , None , context_value = Context (), executor = executor )
137134 assert not result .errors
138135 assert result .data == {
@@ -149,5 +146,5 @@ class Context(object):
149146 }
150147 },
151148 }
152- assert business_load_calls == [['1' ,'2' ]]
153- assert location_load_calls == [['location-1' ,'location-2' ]]
149+ assert business_load_calls == [['1' , '2' ]]
150+ assert location_load_calls == [['location-1' , 'location-2' ]]
0 commit comments