def test_ensure_future_on_awaitable(loop, future):
task = aio.ensure_future(_Awaitable(future))
assert isinstance(task, aio.Task)
loop.call_soon(future.set_result, 9)
assert loop.run_until_complete(task) == 9
def test_ensure_future_on_coroutine(loop):
task = aio.ensure_future(_coro_returning(9))
assert isinstance(task, aio.Task)
assert loop.run_until_complete(task) == 9
def test_ensure_future_on_int():
with pytest.raises(TypeError):
aio.ensure_future(9)
def test_coroutine_with_task(loop):
inner_task = aio.ensure_future(_coro_returning(9))
task = aio.Task(_wait(inner_task))
assert loop.run_until_complete(task) == 9
def test_ensure_future_on_future(future):
assert aio.ensure_future(future) is future