Author: Amaury Forgeot d'Arc <[email protected]>
Branch: py3.6
Changeset: r93955:df41ee03b376
Date: 2018-03-04 14:52 +0100
http://bitbucket.org/pypy/pypy/changeset/df41ee03b376/
Log: Move coroutine tests to test_coroutine.py
diff --git a/pypy/interpreter/astcompiler/test/test_compiler.py
b/pypy/interpreter/astcompiler/test/test_compiler.py
--- a/pypy/interpreter/astcompiler/test/test_compiler.py
+++ b/pypy/interpreter/astcompiler/test/test_compiler.py
@@ -1466,55 +1466,3 @@
return f'ab{x}cd'
"""
code, blocks = generate_function_code(source, self.space)
-
-class AppTestCoroutine:
- def w_run_async(self, coro):
- buffer = []
- result = None
- while True:
- try:
- buffer.append(coro.send(None))
- except StopIteration as ex:
- result = ex.args[0] if ex.args else None
- break
- return buffer, result
-
- def test_async_generator(self):
- """
- async def f(i):
- return i
-
- async def run_list():
- return [await c for c in [f(1), f(41)]]
-
- assert self.run_async(run_list()) == ([], [1, 41])
- """
-
- def test_async_genexpr(self):
- """
- async def f(it):
- for i in it:
- yield i
-
- async def run_gen():
- gen = (i + 1 async for i in f([10, 20]))
- return [g + 100 async for g in gen]
-
- assert self.run_async(run_gen()) == ([], [111, 121])
- """
-
- def test_anext_tuple(self):
- """
- async def foo():
- try:
- yield (1,)
- except ZeroDivisionError:
- yield (2,)
-
- async def run():
- it = foo().__aiter__()
- return await it.__anext__()
-
- assert self.run_async(run()) == ([], (1,))
- """
-
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
@@ -499,3 +499,53 @@
else:
assert False, "didn't raise"
"""
+
+ def w_run_async(self, coro):
+ buffer = []
+ result = None
+ while True:
+ try:
+ buffer.append(coro.send(None))
+ except StopIteration as ex:
+ result = ex.args[0] if ex.args else None
+ break
+ return buffer, result
+
+ def test_async_generator(self):
+ """
+ async def f(i):
+ return i
+
+ async def run_list():
+ return [await c for c in [f(1), f(41)]]
+
+ assert self.run_async(run_list()) == ([], [1, 41])
+ """
+
+ def test_async_genexpr(self):
+ """
+ async def f(it):
+ for i in it:
+ yield i
+
+ async def run_gen():
+ gen = (i + 1 async for i in f([10, 20]))
+ return [g + 100 async for g in gen]
+
+ assert self.run_async(run_gen()) == ([], [111, 121])
+ """
+
+ def test_anext_tuple(self):
+ """
+ async def foo():
+ try:
+ yield (1,)
+ except ZeroDivisionError:
+ yield (2,)
+
+ async def run():
+ it = foo().__aiter__()
+ return await it.__anext__()
+
+ assert self.run_async(run()) == ([], (1,))
+ """
_______________________________________________
pypy-commit mailing list
[email protected]
https://mail.python.org/mailman/listinfo/pypy-commit
