test_buffer.py 508 B

123456789101112131415161718192021
  1. # coding: utf-8
  2. from pandas.io.msgpack import packb, unpackb
  3. from .common import frombytes
  4. def test_unpack_buffer():
  5. from array import array
  6. buf = array('b')
  7. frombytes(buf, packb((b'foo', b'bar')))
  8. obj = unpackb(buf, use_list=1)
  9. assert [b'foo', b'bar'] == obj
  10. def test_unpack_bytearray():
  11. buf = bytearray(packb(('foo', 'bar')))
  12. obj = unpackb(buf, use_list=1)
  13. assert [b'foo', b'bar'] == obj
  14. expected_type = bytes
  15. assert all(type(s) == expected_type for s in obj)