Fix some test failure.

This commit is contained in:
INADA Naoki 2012-10-01 01:34:58 +09:00
parent e016b3dca0
commit d5f99959cc
2 changed files with 8 additions and 8 deletions

View file

@ -99,9 +99,9 @@ def testArraySize(sizes=[0, 5, 50, 1000]):
bio.write(packer.pack(i))
bio.seek(0)
unpacker = Unpacker(bio)
unpacker = Unpacker(bio, use_list=1)
for size in sizes:
assert unpacker.unpack() == tuple(range(size))
assert unpacker.unpack() == list(range(size))
def testMapSize(sizes=[0, 5, 50, 1000]):
bio = six.BytesIO()
@ -115,7 +115,7 @@ def testMapSize(sizes=[0, 5, 50, 1000]):
bio.seek(0)
unpacker = Unpacker(bio)
for size in sizes:
assert unpacker.unpack() == {i: i * 2 for i in range(size)}
assert unpacker.unpack() == dict((i, i * 2) for i in range(size))

View file

@ -6,9 +6,9 @@ def test_read_array_header():
unpacker = Unpacker()
unpacker.feed(packb(['a', 'b', 'c']))
assert unpacker.read_array_header() == 3
assert unpacker.unpack() == 'a'
assert unpacker.unpack() == 'b'
assert unpacker.unpack() == 'c'
assert unpacker.unpack() == b'a'
assert unpacker.unpack() == b'b'
assert unpacker.unpack() == b'c'
try:
unpacker.unpack()
assert 0, 'should raise exception'
@ -20,8 +20,8 @@ def test_read_map_header():
unpacker = Unpacker()
unpacker.feed(packb({'a': 'A'}))
assert unpacker.read_map_header() == 1
assert unpacker.unpack() == 'a'
assert unpacker.unpack() == 'A'
assert unpacker.unpack() == B'a'
assert unpacker.unpack() == B'A'
try:
unpacker.unpack()
assert 0, 'should raise exception'