def test_infinite():
def infinite():
i = 0
while True:
yield i
i += 1
def infinite_str():
i = 0
while True:
yield str(i)
i += 1
def infinite_tupple():
i = 0
while True:
yield (i, str(i))
i += 1
stream = streams.wrap(infinite())
utils.testInfinite(infinite_tupple(), stream.zip(infinite_str()))
finite = [1, 2, 3]
result = [(1, 1), (2, 2), (3, 3)]
utils.checkFinite(result, streams.wrap(itertools.count(1)).zip(finite))
utils.checkFinite(result, streams.wrap(finite).zip(itertools.count(1)))
def test_size_mismatch():
stream1 = streams.wrap([0, 1, 2, 3, 4])
stream2 = ['0', '1', '2']
result = [(0, '0'), (1, '1'), (2, '2')]
utils.checkFinite(result, stream1.zip(stream2))
stream1 = streams.wrap(iter([0, 1, 2]))
utils.checkFinite(result, stream1.zip(stream2 + [4]))
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_basic():
stream = streams.wrap(iter([0, 1, 2]))
utils.checkFinite(['0', '1', '2'], stream.map(str))
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_basic():
sample_list = [[1, 2], [3, 4]]
stream = streams.wrap(iter(sample_list))
utils.checkFinite([1, 2, 3, 4], stream.flatten())
def test_infinite():
stream = streams.wrap(utils.infinte())
utils.checkFinite(['a', 'a', 'a', 'a'], 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_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_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_basic():
sample_list = [1, 2, 3]
stream = streams.wrap(iter(sample_list))
utils.checkFinite(sample_list, stream)
