29 lines
434 B
Python
29 lines
434 B
Python
import pytest
|
|
|
|
from zvk.util.zlogging import logger
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_gen():
|
|
async def f1():
|
|
logger.info('hi 1')
|
|
|
|
async def f2():
|
|
logger.info('hi 2')
|
|
yield 1
|
|
|
|
async def f3():
|
|
logger.info('hi 3')
|
|
if False:
|
|
yield 1
|
|
|
|
with pytest.raises(Exception):
|
|
async for i in f1():
|
|
assert i == 1
|
|
|
|
async for i in f2():
|
|
assert i == 1
|
|
|
|
async for i in f3():
|
|
assert i == 1
|