]> granicus.if.org Git - python/commitdiff
asyncio: Fix two "Coroutine xxx was never yielded from" messages in tests
authorVictor Stinner <victor.stinner@gmail.com>
Fri, 27 Jun 2014 23:19:11 +0000 (01:19 +0200)
committerVictor Stinner <victor.stinner@gmail.com>
Fri, 27 Jun 2014 23:19:11 +0000 (01:19 +0200)
Lib/test/test_asyncio/test_tasks.py

index dee14b2e0a5f2873079634611a66862aade567a1..b4a3092eaf8970f0fce369994f7e5fcc44eeadb0 100644 (file)
@@ -1423,8 +1423,10 @@ class TaskTests(test_utils.TestCase):
         # as_completed() expects a list of futures, not a future instance
         self.assertRaises(TypeError, self.loop.run_until_complete,
             asyncio.as_completed(fut, loop=self.loop))
+        coro = coroutine_function()
         self.assertRaises(TypeError, self.loop.run_until_complete,
-            asyncio.as_completed(coroutine_function(), loop=self.loop))
+            asyncio.as_completed(coro, loop=self.loop))
+        coro.close()
 
     def test_wait_invalid_args(self):
         fut = asyncio.Future(loop=self.loop)
@@ -1432,8 +1434,10 @@ class TaskTests(test_utils.TestCase):
         # wait() expects a list of futures, not a future instance
         self.assertRaises(TypeError, self.loop.run_until_complete,
             asyncio.wait(fut, loop=self.loop))
+        coro = coroutine_function()
         self.assertRaises(TypeError, self.loop.run_until_complete,
-            asyncio.wait(coroutine_function(), loop=self.loop))
+            asyncio.wait(coro, loop=self.loop))
+        coro.close()
 
         # wait() expects at least a future
         self.assertRaises(ValueError, self.loop.run_until_complete,