[pypy-commit] pypy py3.5-corowrapper: Direct test for 'await'

[arigo]

Author: Armin Rigo <arigo at tunes.org>
Branch: py3.5-corowrapper
Changeset: r87196:3821014886f7
Date: 2016-09-17 22:25 +0200
http://bitbucket.org/pypy/pypy/changeset/3821014886f7/

Log:	Direct test for 'await'

diff --git a/pypy/interpreter/test/test_coroutine.py b/pypy/interpreter/test/test_coroutine.py
--- a/pypy/interpreter/test/test_coroutine.py
+++ b/pypy/interpreter/test/test_coroutine.py
@@ -25,7 +25,7 @@
             return sum
         cr = f(X())
         try:
-            next(cr.__await__())
+            cr.send(None)
         except StopIteration as e:
             assert e.value == 42 * 3
         else:
@@ -79,10 +79,28 @@
                 return 42
         c = f(X())
         try:
-            next(c.__await__())
+            c.send(None)
         except StopIteration as e:
             assert e.value == 42
         else:
             assert False, "should have raised"
         assert seen == ['aenter', 'aexit']
         """
+
+    def test_await(self): """
+        class X:
+            def __await__(self):
+                i1 = yield 40
+                assert i1 == 82
+                i2 = yield 41
+                assert i2 == 93
+        async def f():
+            await X()
+            await X()
+        c = f()
+        assert c.send(None) == 40
+        assert c.send(82) == 41
+        assert c.send(93) == 40
+        assert c.send(82) == 41
+        raises(StopIteration, c.send, 93)
+        """