示例#1
0
    def test_compress(self):
        it = itertools.compress(['a', 'b', 'c'], [0, 1, 0])

        assert list(it) == ['b']
示例#2
0
 def test_compress_diff_len(self):
     it = itertools.compress(['a'], [])
     raises(StopIteration, it.next)