def test_compress(self): it = itertools.compress(['a', 'b', 'c'], [0, 1, 0]) assert list(it) == ['b']
def test_compress_diff_len(self): it = itertools.compress(['a'], []) raises(StopIteration, it.next)