Пример #1
0
def test_readline_with_size():
    source = (b'Hello, world! This is a short village name in Wales.\n'
              b'Llanfairpwllgwyngyllgogerychwyrndrobwllllantysiliogogogoch')

    stream = BufferedReader(io.BytesIO(source).read, len(source))
    assert stream.readline(37) == b'Hello, world! This is a short village'
    assert stream.readline(37) == b' name in Wales.\n'
    assert stream.readline(8) == b'Llanfair'
    assert stream.readline(16) == b'pwllgwyngyllgoge'
    assert stream.readline(64) == b'rychwyrndrobwllllantysiliogogogoch'
Пример #2
0
def test_readline():
    source = (b'Hello, world!\n'
              b'A line.\n'
              b'\n'
              b'A longer line... \n' + b'SPAM ' * 7 + b'\n' + b'\n')

    stream = BufferedReader(io.BytesIO(source).read, len(source))
    assert stream.readline() == b'Hello, world!\n'
    assert stream.readline() == b'A line.\n'
    assert stream.readline() == b'\n'
    assert stream.readline() == b'A longer line... \n'
    assert stream.readline() == b'SPAM SPAM SPAM SPAM SPAM SPAM SPAM \n'
    assert stream.readline() == b'\n'
    assert stream.readline() == b''
    assert stream.readline() == b''