def test_async_gen_no_flag_is_async_False(self):
# Test inference when we request a non-async function
# Infer generator
code = ConcreteBytecode()
code.append(ConcreteInstr("YIELD_VALUE"))
code.flags = CompilerFlags(CompilerFlags.COROUTINE)
code.update_flags(is_async=False)
self.assertTrue(bool(code.flags & CompilerFlags.GENERATOR))
# Abort on coroutine
code = ConcreteBytecode()
code.append(ConcreteInstr("GET_AWAITABLE"))
code.flags = CompilerFlags(CompilerFlags.COROUTINE)
with self.assertRaises(ValueError):
code.update_flags(is_async=False)
def test_async_gen_no_flag_is_async_True(self):
# Test inference when we request an async function
# Force coroutine
code = ConcreteBytecode()
code.update_flags(is_async=True)
self.assertTrue(bool(code.flags & CompilerFlags.COROUTINE))
# Infer coroutine or async generator
for i, expected in (
("YIELD_VALUE", CompilerFlags.ASYNC_GENERATOR),
("YIELD_FROM", CompilerFlags.COROUTINE),
):
code = ConcreteBytecode()
code.append(ConcreteInstr(i))
code.update_flags(is_async=True)
self.assertTrue(bool(code.flags & expected))
def test_async_gen_no_flag_is_async_None(self):
# Test inference in the absence of any flag set on the bytecode
# Infer generator
code = ConcreteBytecode()
code.append(ConcreteInstr("YIELD_VALUE"))
code.update_flags()
self.assertTrue(bool(code.flags & CompilerFlags.GENERATOR))
# Infer coroutine
code = ConcreteBytecode()
code.append(ConcreteInstr("GET_AWAITABLE"))
code.update_flags()
self.assertTrue(bool(code.flags & CompilerFlags.COROUTINE))
# Infer coroutine or async generator
for i, expected in (
("YIELD_VALUE", CompilerFlags.ASYNC_GENERATOR),
("YIELD_FROM", CompilerFlags.COROUTINE),
):
code = ConcreteBytecode()
code.append(ConcreteInstr("GET_AWAITABLE"))
code.append(ConcreteInstr(i))
code.update_flags()
self.assertTrue(bool(code.flags & expected))
def test_async_gen_flags(self):
# Test inference in the presence of pre-existing flags
for is_async in (None, True):
# Infer generator
code = ConcreteBytecode()
code.append(ConcreteInstr("YIELD_VALUE"))
for f, expected in (
(CompilerFlags.COROUTINE, CompilerFlags.ASYNC_GENERATOR),
(CompilerFlags.ASYNC_GENERATOR, CompilerFlags.ASYNC_GENERATOR),
(CompilerFlags.ITERABLE_COROUTINE, CompilerFlags.ITERABLE_COROUTINE),
):
code.flags = CompilerFlags(f)
code.update_flags(is_async=is_async)
self.assertTrue(bool(code.flags & expected))
# Infer coroutine
code = ConcreteBytecode()
code.append(ConcreteInstr("YIELD_FROM"))
for f, expected in (
(CompilerFlags.COROUTINE, CompilerFlags.COROUTINE),
(CompilerFlags.ASYNC_GENERATOR, CompilerFlags.COROUTINE),
(CompilerFlags.ITERABLE_COROUTINE, CompilerFlags.ITERABLE_COROUTINE),
):
code.flags = CompilerFlags(f)
code.update_flags(is_async=is_async)
self.assertTrue(bool(code.flags & expected))
# Crash on ITERABLE_COROUTINE with async bytecode
code = ConcreteBytecode()
code.append(ConcreteInstr("GET_AWAITABLE"))
code.flags = CompilerFlags(CompilerFlags.ITERABLE_COROUTINE)
with self.assertRaises(ValueError):
code.update_flags(is_async=is_async)
