Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2025-11-09 14:01:03 +00:00
Code Issues Projects Releases Packages Wiki Activity

Packer.pack_array/map_header to correspond to read functions

Browse source
This commit is contained in:
Joel Nothman 2012-09-23 17:26:16 +10:00
parent 0431a766f4
commit 9d9c3eecb8
2 changed files with 40 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

29
test/test_pack.py
View file

@ -91,6 +91,35 @@ def testPackFloat():
assert_equal(packb(1.0, use_single_float=True), b'\xca' + struct.pack('>f', 1.0))
assert_equal(packb(1.0, use_single_float=False), b'\xcb' + struct.pack('>d', 1.0))
def testArraySize(sizes=[0, 5, 50, 1000]):
bio = six.BytesIO()
packer = Packer()
for size in sizes:
bio.write(packer.pack_array_header(size))
for i in range(size):
bio.write(packer.pack(i))
bio.seek(0)
unpacker = Unpacker(bio)
for size in sizes:
assert unpacker.unpack() == tuple(range(size))
def testMapSize(sizes=[0, 5, 50, 1000]):
bio = six.BytesIO()
packer = Packer()
for size in sizes:
bio.write(packer.pack_map_header(size))
for i in range(size):
bio.write(packer.pack(i)) # key
bio.write(packer.pack(i * 2)) # value
bio.seek(0)
unpacker = Unpacker(bio)
for size in sizes:
assert unpacker.unpack() == {i: i * 2 for i in range(size)}
class odict(dict):
'''Reimplement OrderedDict to run test on Python 2.6'''