python
diff --git a/‎Lib/asyncio/tasks.py
Copy file name to clipboardExpand all lines: Lib/asyncio/tasks.py
+4-2Lines changed: 4 additions & 2 deletions b/‎Lib/asyncio/tasks.py
Copy file name to clipboardExpand all lines: Lib/asyncio/tasks.py
+4-2Lines changed: 4 additions & 2 deletions
diff --git a/‎Lib/test/test_asyncio/test_tasks.py
Copy file name to clipboardExpand all lines: Lib/test/test_asyncio/test_tasks.py
+24Lines changed: 24 additions & 0 deletions b/‎Lib/test/test_asyncio/test_tasks.py
Copy file name to clipboardExpand all lines: Lib/test/test_asyncio/test_tasks.py
+24Lines changed: 24 additions & 0 deletions
diff --git a/‎Misc/NEWS.d/next/Library/2020-10-24-04-02-36.bpo-42140.miLqvb.rst
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎Misc/NEWS.d/next/Library/2020-10-24-04-02-36.bpo-42140.miLqvb.rst
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
@@ -400,13 +400,15 @@ async def wait(fs, *, loop=None, timeout=None, return_when=ALL_COMPLETED):
                       "and scheduled for removal in Python 3.10.",
                       DeprecationWarning, stacklevel=2)
 
-    if any(coroutines.iscoroutine(f) for f in set(fs)):
+    fs = set(fs)
+
+    if any(coroutines.iscoroutine(f) for f in fs):
         warnings.warn("The explicit passing of coroutine objects to "
                       "asyncio.wait() is deprecated since Python 3.8, and "
                       "scheduled for removal in Python 3.11.",
                       DeprecationWarning, stacklevel=2)
 
-    fs = {ensure_future(f, loop=loop) for f in set(fs)}
+    fs = {ensure_future(f, loop=loop) for f in fs}
 
     return await _wait(fs, timeout, return_when, loop)
 
 
@@ -1548,6 +1548,30 @@ def gen():
         loop.advance_time(10)
         loop.run_until_complete(asyncio.wait([a, b]))
 
+    def test_wait_with_iterator_of_tasks(self):
+
+        def gen():
+            when = yield
+            self.assertAlmostEqual(0.1, when)
+            when = yield 0
+            self.assertAlmostEqual(0.15, when)
+            yield 0.15
+
+        loop = self.new_test_loop(gen)
+
+        a = self.new_task(loop, asyncio.sleep(0.1))
+        b = self.new_task(loop, asyncio.sleep(0.15))
+
+        async def foo():
+            done, pending = await asyncio.wait(iter([b, a]))
+            self.assertEqual(done, set([a, b]))
+            self.assertEqual(pending, set())
+            return 42
+
+        res = loop.run_until_complete(self.new_task(loop, foo()))
+        self.assertEqual(res, 42)
+        self.assertAlmostEqual(0.15, loop.time())
+
     def test_as_completed(self):
 
         def gen():
 
@@ -0,0 +1 @@
+Improve asyncio.wait function to create the futures set just one time.
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+Improve asyncio.wait function to create the futures set just one time.`