示例#1
0
def test_streamer_infinite(n_max, stream_size):
    reference = []
    for i, data in enumerate(T.infinite_generator(size=stream_size)):
        if i >= n_max:
            break
        reference.append(data)

    streamer = pescador.core.Streamer(T.infinite_generator, size=stream_size)

    for i in range(3):
        query = list(streamer.iterate(max_iter=n_max))

        for b1, b2 in zip(reference, query):
            T._eq_batch(b1, b2)
示例#2
0
def test_streamer_infinite(n_max, stream_size):
    reference = []
    for i, data in enumerate(T.infinite_generator(size=stream_size)):
        if i >= n_max:
            break
        reference.append(data)

    streamer = pescador.core.Streamer(T.infinite_generator, size=stream_size)

    for i in range(3):
        query = list(streamer.iterate(max_iter=n_max))

        for b1, b2 in zip(reference, query):
            T._eq_batch(b1, b2)
示例#3
0
def test_streamer_in_streamer(n_max, stream_size):
    # TODO minimize copypasta from above test.
    reference = []
    for i, data in enumerate(T.infinite_generator(size=stream_size)):
        if i >= n_max:
            break
        reference.append(data)

    streamer = pescador.core.Streamer(T.infinite_generator, size=stream_size)

    streamer2 = pescador.core.Streamer(streamer)

    for i in range(3):
        query = list(streamer2.iterate(max_iter=n_max))

        for b1, b2 in zip(reference, query):
            T.__eq_batch(b1, b2)