mirror of
https://github.com/python/cpython.git
synced 2025-08-25 03:04:55 +00:00
Fix wrapping into StopIteration of return values in generators and coroutines (#644)
This commit is contained in:
parent
2b27e2e6a3
commit
b7c9150b68
2 changed files with 16 additions and 2 deletions
|
@ -1103,6 +1103,21 @@ class CoroutineTest(unittest.TestCase):
|
||||||
"coroutine is being awaited already"):
|
"coroutine is being awaited already"):
|
||||||
waiter(coro).send(None)
|
waiter(coro).send(None)
|
||||||
|
|
||||||
|
def test_await_16(self):
|
||||||
|
# See https://bugs.python.org/issue29600 for details.
|
||||||
|
|
||||||
|
async def f():
|
||||||
|
return ValueError()
|
||||||
|
|
||||||
|
async def g():
|
||||||
|
try:
|
||||||
|
raise KeyError
|
||||||
|
except:
|
||||||
|
return await f()
|
||||||
|
|
||||||
|
_, result = run_async(g())
|
||||||
|
self.assertIsNone(result.__context__)
|
||||||
|
|
||||||
def test_with_1(self):
|
def test_with_1(self):
|
||||||
class Manager:
|
class Manager:
|
||||||
def __init__(self, name):
|
def __init__(self, name):
|
||||||
|
|
|
@ -574,8 +574,7 @@ _PyGen_SetStopIterationValue(PyObject *value)
|
||||||
PyObject *e;
|
PyObject *e;
|
||||||
|
|
||||||
if (value == NULL ||
|
if (value == NULL ||
|
||||||
(!PyTuple_Check(value) &&
|
(!PyTuple_Check(value) && !PyExceptionInstance_Check(value)))
|
||||||
!PyObject_TypeCheck(value, (PyTypeObject *) PyExc_StopIteration)))
|
|
||||||
{
|
{
|
||||||
/* Delay exception instantiation if we can */
|
/* Delay exception instantiation if we can */
|
||||||
PyErr_SetObject(PyExc_StopIteration, value);
|
PyErr_SetObject(PyExc_StopIteration, value);
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue