diff options
Diffstat (limited to 'Lib/test/test_asyncio/test_queues.py')
| -rw-r--r-- | Lib/test/test_asyncio/test_queues.py | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/Lib/test/test_asyncio/test_queues.py b/Lib/test/test_asyncio/test_queues.py index 0a0b529f621..63a9a5f270c 100644 --- a/Lib/test/test_asyncio/test_queues.py +++ b/Lib/test/test_asyncio/test_queues.py @@ -273,12 +273,12 @@ class QueueGetTests(_QueueTestBase): queue._get_loop() return queue - q = self.loop.run_until_complete(create_queue()) + async def test(): + q = await create_queue() + await asyncio.gather(producer(q, producer_num_items), + consumer(q, producer_num_items)) - self.loop.run_until_complete( - asyncio.gather(producer(q, producer_num_items), - consumer(q, producer_num_items)), - ) + self.loop.run_until_complete(test()) def test_cancelled_getters_not_being_held_in_self_getters(self): def a_generator(): @@ -516,11 +516,14 @@ class QueuePutTests(_QueueTestBase): for _ in range(num): item = queue.get_nowait() - t0 = putter(0) - t1 = putter(1) - t2 = putter(2) - t3 = putter(3) - self.loop.run_until_complete(asyncio.gather(getter(), t0, t1, t2, t3)) + async def test(): + t0 = putter(0) + t1 = putter(1) + t2 = putter(2) + t3 = putter(3) + await asyncio.gather(getter(), t0, t1, t2, t3) + + self.loop.run_until_complete(test()) def test_cancelled_puts_not_being_held_in_self_putters(self): def a_generator(): |