def test_infinite(): stream = streams.wrap(itertools.count()) if streams._py_major == 2 and streams._py_minor < 7: def count(start, step): while True: yield start start += step else: count = itertools.count utils.testInfinite(count(1, 2), stream.filter(lambda x: x % 2 == 1))
def test_exact_page(): stream = streams.wrap(iter([0, 1, 2, 3])).paginate(2) layers = 0 for l in stream: utils.testWrapped(l) values = 0 for v in l: assert v == values + (2 * layers) values += 1 assert values == 2 layers += 1 assert layers == 2 utils.testEmpty(stream)
def test_infinite(): def infinte(): i = 0 while True: yield i i += 1 def infinte_str(): i = 0 while True: yield str(i) i += 1 stream = streams.wrap(infinte()) utils.testInfinite(infinte_str(), stream.map(str))
def test_lazy(): size = 4 stream = streams.wrap(utils.lazy_bomb(size / 2, [1, 2])).flatten() for i in range(size): next(stream)
def test_basic(): stream = streams.wrap(iter([0, 1, 2])) utils.checkFinite(['0', '1', '2'], stream.map(str))
def test_infinite(): stream = streams.wrap(utils.infinte()) utils.testInfinite(utils.infinte(), stream)
def test_empty(): stream = streams.wrap(iter([])) utils.testEmpty(stream.flatten())
def test_basic(): sample_list = [1, 2, 3] stream = streams.wrap(iter(sample_list)) utils.checkFinite([1, 3], stream.filter(lambda x: x % 2 == 1))
def test_multi(): stream1 = streams.wrap(iter([0, 1, 2])) stream2 = ['0', '1', '2'] stream3 = ["x", "y", "z"] result = [(0, '0', "x"), (1, '1', "y"), (2, '2', "z")] utils.checkFinite(result, stream1.zip(stream2, stream3))
def test_zero(): sample_list = [1, 2, 3, 4, 5] stream = streams.wrap(iter(sample_list)) utils.testEmpty(stream.limit(0))
def test_lazy(): size = 4 stream = streams.wrap(utils.lazy_bomb(size)).limit(5) for i in range(size): next(stream)
def test_infinite(): stream = streams.wrap(utils.infinte()) utils.checkFinite(['a', 'a', 'a', 'a'], stream.limit(4))
def test_empty(): stream = streams.wrap(iter([])) utils.testEmpty(stream.limit(4))
def test_high_limit(): sample_list = [1, 2, 3, 4, 5] stream = streams.wrap(iter(sample_list)) utils.checkFinite(sample_list, stream.limit(6))
def test_basic(): sample_list = [1, 2, 3, 4, 5] stream = streams.wrap(iter(sample_list)) utils.checkFinite(sample_list[:4], stream.limit(4))
def test_empty(): stream = streams.wrap(iter([])) finite = streams.wrap([1, 2, 3]) utils.testEmpty(stream.zip(finite)) utils.testEmpty(finite.zip(stream))
def test_empty(): stream = streams.wrap(iter([])) utils.testEmpty(stream.map(str))
def test_empty(): stream = streams.wrap(iter([])) utils.testEmpty(stream.filter(lambda x: x % 2 == 1))
def test_lazy(): size = 4 stream = streams.wrap(utils.lazy_bomb(size)).map(lambda x: x) for i in range(size): next(stream)
def test_empty(): stream = streams.wrap(iter([])).paginate(1) utils.testEmpty(stream)
def test_basic(): sample_list = [[1, 2], [3, 4]] stream = streams.wrap(iter(sample_list)) utils.checkFinite([1, 2, 3, 4], stream.flatten())
def test_basic(): stream1 = streams.wrap(iter([0, 1, 2])) stream2 = ['0', '1', '2'] result = [(0, '0'), (1, '1'), (2, '2')] utils.checkFinite(result, stream1.zip(stream2))
def test_lazy(): size = 4 stream = streams.wrap(utils.lazy_bomb(size)).filter(lambda x: True) for i in range(size): next(stream)
def test_recursive(): sample_list = [streams.wrap([1, 2]), streams.wrap([3])] stream = streams.wrap(iter(sample_list)) utils.checkFinite([1, 2, 3], stream.flatten())
def test_infinite(): stream = streams.wrap(utils.infinte(['a'])) utils.testInfinite(utils.infinte(), stream.flatten()) stream = streams.wrap([utils.infinte(), utils.infinte()]) utils.testInfinite(utils.infinte(), stream.flatten())
def test_basic(): sample_list = [1, 2, 3] stream = streams.wrap(iter(sample_list)) utils.checkFinite(sample_list, stream)