async def test_asyncio_req_generator_empty_inputs():
generator = request_generator('/', None)
i = 0
async for req in generator:
i += 1
assert len(req.docs) == 0
await asyncio.sleep(0.1)
assert i == 1
async def test_asyncio_req_generator():
async def input_function():
data = [Document() for _ in range(NUM_INPUT_DOCS)]
for doc in data:
yield doc
generator = request_generator('/', input_function(), request_size=REQUEST_SIZE)
i = 0
async for req in generator:
i += 1
assert len(req.docs) == REQUEST_SIZE
await asyncio.sleep(0.1)
assert i == ceil(NUM_INPUT_DOCS / REQUEST_SIZE)
def test_asyncio_bad_input_generator():
# exception not handled
data = ['text' for _ in range(20)]
request_generator('/', data, request_size=10)
def test_asyncio_bad_input_generator():
# exception not handled
data = ['text' for _ in range(20)]
request_generator(data, request_size=10, mode='not_index')