msgpack-python/test/test_buffer.py
2019-01-25 20:52:57 +09:00

32 lines
836 B
Python

#!/usr/bin/env python
# coding: utf-8
from msgpack import packb, unpackb
def test_unpack_buffer():
from array import array
buf = array('b')
try:
buf.frombytes(packb((b'foo', b'bar')))
except AttributeError: # PY2
buf.fromstring(packb((b'foo', b'bar')))
obj = unpackb(buf, use_list=1)
assert [b'foo', b'bar'] == obj
def test_unpack_bytearray():
buf = bytearray(packb(('foo', 'bar')))
obj = unpackb(buf, use_list=1)
assert [b'foo', b'bar'] == obj
expected_type = bytes
assert all(type(s) == expected_type for s in obj)
def test_unpack_memoryview():
buf = bytearray(packb(('foo', 'bar')))
view = memoryview(buf)
obj = unpackb(view, use_list=1)
assert [b'foo', b'bar'] == obj
expected_type = bytes
assert all(type(s) == expected_type for s in obj)