| 123456789101112131415161718192021222324252627282930 |
- """Tests for cases where the user seeks to obtain packed msgpack objects"""
- import io
- from pandas.io.msgpack import Unpacker, packb
- def test_write_bytes():
- unpacker = Unpacker()
- unpacker.feed(b'abc')
- f = io.BytesIO()
- assert unpacker.unpack(f.write) == ord('a')
- assert f.getvalue() == b'a'
- f = io.BytesIO()
- assert unpacker.skip(f.write) is None
- assert f.getvalue() == b'b'
- f = io.BytesIO()
- assert unpacker.skip() is None
- assert f.getvalue() == b''
- def test_write_bytes_multi_buffer():
- long_val = (5) * 100
- expected = packb(long_val)
- unpacker = Unpacker(io.BytesIO(expected), read_size=3, max_buffer_size=3)
- f = io.BytesIO()
- unpacked = unpacker.unpack(f.write)
- assert unpacked == long_val
- assert f.getvalue() == expected
|
