def test_wrong_chainging_raises_error():
@push_map
def square(x):
return x * x
with raises(TypeError):
pull_from(range(10)) >> square
def test_broadcast():
result1 = []
result2 = []
sink = broadcast(sink_map(result1.append), sink_map(result2.append))
patch(pull_from(range(10)), sink)
assert result1 == result2 == list(range(10))
def test_queue():
Q = Queue()
patch(pull_from(range(10)) >> (lambda x: x * x), Q.sink)
Q.close()
assert list(Q.source) == [i * i for i in range(10)]
with raises(StopIteration):
next(Q.source())
def test_push_chaining():
def square(x):
return x * x
squares = []
patch(pull_from(range(10)), push_map(square) >> sink_map(squares.append))
assert squares == [i**2 for i in range(10)]
def test_pull_mapping():
@pull_map
def square(x):
return x * x
squares = pull_from(range(10)) >> square
assert list(squares) == [i**2 for i in range(10)]
def test_queue_chaining():
Q = Queue() >> (lambda x: x * x)
patch(pull_from(range(10)), Q.sink)
try:
Q.sink().send(EndOfQueue)
except StopIteration:
pass
assert list(Q.source) == [i * i for i in range(10)]
def test_pull_chaining():
@pull
def square(source):
for x in source():
yield x * x
squares = pull_from(range(10)) >> square
assert list(squares) == [i**2 for i in range(10)]
def test_branch():
squares = []
cubes = []
square = push_map(lambda x: x**2) >> sink_map(squares.append)
cube = push_map(lambda x: x**3) >> sink_map(cubes.append)
numbers = list(pull_from(range(10)) >> branch(square, cube))
assert numbers == list(range(10))
assert cubes == [i**3 for i in range(10)]
assert squares == [i**2 for i in range(10)]
def test_thread_pool():
@pull_map
def square(x):
return x * x
Q = Queue()
worker = Q >> thread_pool(square, square)
patch(pull_from(range(10)), worker.sink)
Q.close()
assert sorted(list(worker.source)) == [i**2 for i in range(10)]
def test_function_chaining():
squares = pull_from(range(10)) >> (lambda x: x * x)
assert list(squares) == [i**2 for i in range(10)]