def test_async_all_done_pre_flag():
pool = CompletionOrderedAsyncWorkPool()
done = pool.async_all_done()
cirq.testing.assert_asyncio_still_running(done)
pool.set_all_work_received_flag()
cirq.testing.assert_asyncio_will_have_result(done, None)
async def test_async_all_done_pre_flag():
pool = CompletionOrderedAsyncWorkPool()
done = pool.async_all_done()
assert await cirq.testing.asynchronous.asyncio_pending(done)
pool.set_all_work_received_flag()
assert await done is None
def test_async_all_done_flag_then_finish_work():
pool = CompletionOrderedAsyncWorkPool()
done = pool.async_all_done()
work = asyncio.Future()
pool.include_work(work)
pool.set_all_work_received_flag()
cirq.testing.assert_asyncio_still_running(done)
work.set_result(5)
cirq.testing.assert_asyncio_will_have_result(done, None)
async def test_async_all_done_flag_then_finish_work():
pool = CompletionOrderedAsyncWorkPool()
done = pool.async_all_done()
work = asyncio.Future()
pool.include_work(work)
pool.set_all_work_received_flag()
assert await cirq.testing.asynchronous.asyncio_pending(done)
work.set_result(5)
assert await done is None
def test_async_all_done_post_flag():
pool = CompletionOrderedAsyncWorkPool()
pool.set_all_work_received_flag()
cirq.testing.assert_asyncio_will_have_result(pool.async_all_done(), None)
cirq.testing.assert_asyncio_will_have_result(pool.async_all_done(), None)