Change the way to run async function

affjljoo3581 · Jul 6, 2020 · de8efd4 · de8efd4
1 parent 4842836
commit de8efd4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -12,7 +12,10 @@ async def test_fn(i):
         return i
 
     ctx = Context(max_tasks=10)
-    result, errors = asyncio.run(ctx.run(test_fn(i) for i in range(100)))
+
+    loop = asyncio.get_event_loop()
+    result, errors = loop.run_until_complete(
+        ctx.run(test_fn(i) for i in range(100)))
 
     assert errors == 0
     assert set(result) == set(range(100))