Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2025-10-19 20:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add Packer.buffer() (#320)

Browse source
This commit is contained in:
INADA Naoki 2018-11-09 20:55:13 +09:00 • committed by GitHub
parent a8b3e97fe5
commit 9e210bfc1a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 107 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

12
test/test_pack.py
View file

@ -5,7 +5,7 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import struct
from pytest import raises, xfail
from msgpack import packb, unpackb, Unpacker, Packer
from msgpack import packb, unpackb, Unpacker, Packer, pack
from collections import OrderedDict
from io import BytesIO
@ -148,3 +148,13 @@ def test_pairlist():
packed = packer.pack_map_pairs(pairlist)
unpacked = unpackb(packed, object_pairs_hook=list)
assert pairlist == unpacked
def test_get_buffer():
packer = Packer(autoreset=0, use_bin_type=True)
packer.pack([1, 2])
strm = BytesIO()
strm.write(packer.getbuffer())
written = strm.getvalue()
expected = packb([1, 2], use_bin_type=True)
assert written == expected