def test_subpipe_should_return_expected_output(self):
a = Mock(output=lambda: list('abc'))
b = Mock(output=lambda: list('def'))
c = Mock(output=lambda: list('beg'))
subpipe = BasePipe()
subpipe.add(a, b)
pipe = BasePipe()
pipe.add(c, subpipe) # should not raise
self.assertEqual(pipe.output(), subpipe.output())
def test_output_returns_unique_combined_list_from_all_if_passed_true(self):
pipe = BasePipe()
a = Mock(output=lambda: list('abc'))
b = Mock(output=lambda: list('def'))
c = Mock(output=lambda: list('beg'))
pipe.add(a, b, c)
self.assertEqual(pipe.output(from_all=True), list('abcdefg'))
