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.write(packer.pack(i))
bio.seek(0) bio.seek(0)
unpacker = Unpacker(bio) unpacker = Unpacker(bio, use_list=1)
for size in sizes: for size in sizes:
assert unpacker.unpack() == tuple(range(size)) assert unpacker.unpack() == list(range(size))
def testMapSize(sizes=[0, 5, 50, 1000]): def testMapSize(sizes=[0, 5, 50, 1000]):
bio = six.BytesIO() bio = six.BytesIO()
@ -115,7 +115,7 @@ def testMapSize(sizes=[0, 5, 50, 1000]):
bio.seek(0) bio.seek(0)
unpacker = Unpacker(bio) unpacker = Unpacker(bio)
for size in sizes: 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 = Unpacker()
unpacker.feed(packb(['a', 'b', 'c'])) unpacker.feed(packb(['a', 'b', 'c']))
assert unpacker.read_array_header() == 3 assert unpacker.read_array_header() == 3
assert unpacker.unpack() == 'a' assert unpacker.unpack() == b'a'
assert unpacker.unpack() == 'b' assert unpacker.unpack() == b'b'
assert unpacker.unpack() == 'c' assert unpacker.unpack() == b'c'
try: try:
unpacker.unpack() unpacker.unpack()
assert 0, 'should raise exception' assert 0, 'should raise exception'
@ -20,8 +20,8 @@ def test_read_map_header():
unpacker = Unpacker() unpacker = Unpacker()
unpacker.feed(packb({'a': 'A'})) unpacker.feed(packb({'a': 'A'}))
assert unpacker.read_map_header() == 1 assert unpacker.read_map_header() == 1
assert unpacker.unpack() == 'a' assert unpacker.unpack() == B'a'
assert unpacker.unpack() == 'A' assert unpacker.unpack() == B'A'
try: try:
unpacker.unpack() unpacker.unpack()
assert 0, 'should raise exception' assert 0, 'should raise exception'