def test_merge(self, sources, expected_output): in_iters = [iter(src) for src in sources] out = list(utils.merge(10, *in_iters)) self.assertEqual(expected_output, out)