def test_pipe_simple(self):
p = model_util.BufferedPipe() \
.add(lambda x: x + 1) \
.add(lambda x: x * x)
l = list(p((x for x in range(1, 5))))
expected_l = [4, 9, 16, 25]
self.assertEquals(l, expected_l)
def test_pipe_buffer(self):
p = model_util.BufferedPipe() \
.add(lambda x: x + 1, num_workers=32) \
.add(lambda x: x * x)
out_gen = p((x for x in range(1, 5)))
expected_l = [4, 9, 16, 25]
self.assertEquals(list(out_gen), expected_l)
