Make use of async comprehensions

graphql-python · May 16, 2020 · 1948be5 · 1948be5
1 parent e024953
commit 1948be5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/starwars/test_subscription.py b/tests/starwars/test_subscription.py
@@ -50,10 +50,10 @@ async def test_subscription_support_using_client():
     params = {"ep": "JEDI"}
     expected = [{**review, "episode": "JEDI"} for review in reviews[6]]
 
-    results = []
-
     async with Client(schema=StarWarsSchema) as session:
-        async for result in session.subscribe(subs, variable_values=params):
-            results.append(result["reviewAdded"])
+        results = [
+            result["reviewAdded"]
+            async for result in session.subscribe(subs, variable_values=params)
+        ]
 
     assert results == expected