async def test_flat_map_monad_law_associativity():
# (m >>= f) >>= g is just like doing m >>= (\x -> f x >>= g)
m = unit(42)
async def f(x):
return unit(x + 1000)
async def g(y):
return unit(y * 333)
async def h(x):
return flat_map(g, await f(x))
a = await run(flat_map(g, flat_map(f, m)))
b = await run(flat_map(h, m))
assert a == b
async def test_flat_map_monad():
m = unit(42)
async def mapper(x):
return unit(x * 10)
a = await run(flat_map(mapper, m))
b = await run(unit(420))
assert a == b
async def test_flat_map_monad_law_right_identity():
# m >>= return is no different than just m.
m = unit("move on up")
async def aunit(x):
return unit(x)
a = await run(flat_map(aunit, m))
b = await run(m)
assert a == b
async def test_flat_map_monad_law_left_identity():
# return x >>= f is the same thing as f x
x = 3
async def f(x):
return unit(x + 100000)
a = await run(flat_map(f, unit(x)))
b = await run(await f(x))
assert a == b
async def test_flap_map_done():
xs = AsyncStream()
result = []
async def asend(value):
nonlocal result
result.append(value)
async def mapper(value):
return from_iterable([value])
ys = flat_map(mapper, xs)
await subscribe(ys, AnonymousAsyncObserver(asend))
await xs.asend(10)
await xs.asend(20)
await asyncio.sleep(0.6)
assert result == [10, 20]
def select_many(self, selector: Callable) -> 'ChainedAsyncObservable':
from aioreactive.operators.flat_map import flat_map
return ChainedAsyncObservable(flat_map(selector, self))
async def h(x):
return flat_map(g, await f(x))