def test_decompress_stream_readline(): text = six.b("1\n2\n3\n4") s = BytesIO() cstream = compress_stream(s) cstream.write(text) cstream.flush() dstream = decompress_stream(BytesIO(s.getvalue())) assert dstream.readline() == six.b("1\n") assert dstream.readline() == six.b("2\n") assert dstream.readline() == six.b("3\n") assert dstream.readline() == six.b("4") assert dstream.readline() == six.b("")
def test_decompress_stream_readline(): text = "1\n2\n3\n4" s = StringIO() cstream = compress_stream(s) cstream.write(text) cstream.flush() dstream = decompress_stream(StringIO(s.getvalue())) assert dstream.readline() == "1\n" assert dstream.readline() == "2\n" assert dstream.readline() == "3\n" assert dstream.readline() == "4" assert dstream.readline() == ""