AsyncIO experimentation

async.py

import asyncio
import random


async def square(stream):
    async for num in stream:
        yield num ** 2


async def add_two(stream):
    async for num in stream:
        yield num + 2


async def arange(*range_args):
    for x in range(*range_args):
        await asyncio.sleep(random.uniform(0.1, 0.4))
        yield x


async def main():
    values = arange(5)
    values = add_two(values)
    values = square(values)

    return [val async for val in values]


if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    res = loop.run_until_complete(main())
    print('Result:', res)