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)