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'
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''