Fix using deprecated api in tests.

This commit is contained in:
INADA Naoki 2012-07-04 14:58:36 +09:00
parent 4bff55db9f
commit 2122b46b84
5 changed files with 29 additions and 29 deletions

View file

@ -3,13 +3,13 @@
from nose import main from nose import main
from nose.tools import * from nose.tools import *
from msgpack import packs, unpacks from msgpack import packb, unpackb
def check(length, obj): def check(length, obj):
v = packs(obj) v = packb(obj)
assert_equal(len(v), length, "%r length should be %r but get %r" % (obj, length, len(v))) assert_equal(len(v), length, "%r length should be %r but get %r" % (obj, length, len(v)))
assert_equal(unpacks(v), obj) assert_equal(unpackb(v), obj)
def test_1(): def test_1():
for o in [None, True, False, 0, 1, (1 << 6), (1 << 7) - 1, -1, for o in [None, True, False, 0, 1, (1 << 6), (1 << 7) - 1, -1,
@ -70,8 +70,8 @@ def test_array32():
def match(obj, buf): def match(obj, buf):
assert_equal(packs(obj), buf) assert_equal(packb(obj), buf)
assert_equal(unpacks(buf), obj) assert_equal(unpackb(buf), obj)
def test_match(): def test_match():
cases = [ cases = [
@ -99,7 +99,7 @@ def test_match():
match(v, p) match(v, p)
def test_unicode(): def test_unicode():
assert_equal(b'foobar', unpacks(packs('foobar'))) assert_equal(b'foobar', unpackb(packb('foobar')))
if __name__ == '__main__': if __name__ == '__main__':
main() main()

View file

@ -2,12 +2,12 @@
# coding: utf-8 # coding: utf-8
from nose.tools import * from nose.tools import *
from msgpack import packs, unpacks from msgpack import packb, unpackb
import datetime import datetime
def test_raise_on_find_unsupported_value(): def test_raise_on_find_unsupported_value():
assert_raises(TypeError, packs, datetime.datetime.now()) assert_raises(TypeError, packb, datetime.datetime.now())
if __name__ == '__main__': if __name__ == '__main__':
from nose import main from nose import main

View file

@ -3,10 +3,10 @@
from nose import main from nose import main
from nose.tools import * from nose.tools import *
from msgpack import unpacks from msgpack import unpackb
def check(src, should): def check(src, should):
assert_equal(unpacks(src), should) assert_equal(unpackb(src), should)
def testSimpleValue(): def testSimpleValue():
check(b"\x93\xc0\xc2\xc3", check(b"\x93\xc0\xc2\xc3",

View file

@ -4,7 +4,7 @@
from nose import main from nose import main
from nose.tools import * from nose.tools import *
from msgpack import packs, unpacks from msgpack import packb, unpackb
def _decode_complex(obj): def _decode_complex(obj):
if b'__complex__' in obj: if b'__complex__' in obj:
@ -17,26 +17,26 @@ def _encode_complex(obj):
return obj return obj
def test_encode_hook(): def test_encode_hook():
packed = packs([3, 1+2j], default=_encode_complex) packed = packb([3, 1+2j], default=_encode_complex)
unpacked = unpacks(packed) unpacked = unpackb(packed)
eq_(unpacked[1], {b'__complex__': True, b'real': 1, b'imag': 2}) eq_(unpacked[1], {b'__complex__': True, b'real': 1, b'imag': 2})
def test_decode_hook(): def test_decode_hook():
packed = packs([3, {b'__complex__': True, b'real': 1, b'imag': 2}]) packed = packb([3, {b'__complex__': True, b'real': 1, b'imag': 2}])
unpacked = unpacks(packed, object_hook=_decode_complex) unpacked = unpackb(packed, object_hook=_decode_complex)
eq_(unpacked[1], 1+2j) eq_(unpacked[1], 1+2j)
@raises(ValueError) @raises(ValueError)
def test_bad_hook(): def test_bad_hook():
packed = packs([3, 1+2j], default=lambda o: o) packed = packb([3, 1+2j], default=lambda o: o)
unpacked = unpacks(packed) unpacked = unpackb(packed)
def _arr_to_str(arr): def _arr_to_str(arr):
return ''.join(str(c) for c in arr) return ''.join(str(c) for c in arr)
def test_array_hook(): def test_array_hook():
packed = packs([1,2,3]) packed = packb([1,2,3])
unpacked = unpacks(packed, list_hook=_arr_to_str) unpacked = unpackb(packed, list_hook=_arr_to_str)
eq_(unpacked, '123') eq_(unpacked, '123')
if __name__ == '__main__': if __name__ == '__main__':

View file

@ -6,12 +6,12 @@ from nose import main
from nose.tools import * from nose.tools import *
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
from msgpack import packs, unpacks, Unpacker, Packer from msgpack import packb, unpackb, Unpacker, Packer
from io import BytesIO from io import BytesIO
def check(data): def check(data):
re = unpacks(packs(data)) re = unpackb(packb(data))
assert_equal(re, data) assert_equal(re, data)
def testPack(): def testPack():
@ -33,7 +33,7 @@ def testPackUnicode():
six.u(""), six.u("abcd"), (six.u("defgh"),), six.u("Русский текст"), six.u(""), six.u("abcd"), (six.u("defgh"),), six.u("Русский текст"),
] ]
for td in test_data: for td in test_data:
re = unpacks(packs(td, encoding='utf-8'), encoding='utf-8') re = unpackb(packb(td, encoding='utf-8'), encoding='utf-8')
assert_equal(re, td) assert_equal(re, td)
packer = Packer(encoding='utf-8') packer = Packer(encoding='utf-8')
data = packer.pack(td) data = packer.pack(td)
@ -49,7 +49,7 @@ def testPackUTF32():
six.u("Русский текст"), six.u("Русский текст"),
] ]
for td in test_data: for td in test_data:
re = unpacks(packs(td, encoding='utf-32'), encoding='utf-32') re = unpackb(packb(td, encoding='utf-32'), encoding='utf-32')
assert_equal(re, td) assert_equal(re, td)
except LookupError: except LookupError:
raise SkipTest raise SkipTest
@ -62,28 +62,28 @@ def testPackBytes():
check(td) check(td)
def testIgnoreUnicodeErrors(): def testIgnoreUnicodeErrors():
re = unpacks(packs(b'abc\xeddef'), re = unpackb(packb(b'abc\xeddef'),
encoding='utf-8', unicode_errors='ignore') encoding='utf-8', unicode_errors='ignore')
assert_equal(re, "abcdef") assert_equal(re, "abcdef")
@raises(UnicodeDecodeError) @raises(UnicodeDecodeError)
def testStrictUnicodeUnpack(): def testStrictUnicodeUnpack():
unpacks(packs(b'abc\xeddef'), encoding='utf-8') unpackb(packb(b'abc\xeddef'), encoding='utf-8')
@raises(UnicodeEncodeError) @raises(UnicodeEncodeError)
def testStrictUnicodePack(): def testStrictUnicodePack():
packs(six.u("abc\xeddef"), encoding='ascii', unicode_errors='strict') packb(six.u("abc\xeddef"), encoding='ascii', unicode_errors='strict')
def testIgnoreErrorsPack(): def testIgnoreErrorsPack():
re = unpacks(packs(six.u("abcФФФdef"), encoding='ascii', unicode_errors='ignore'), encoding='utf-8') re = unpackb(packb(six.u("abcФФФdef"), encoding='ascii', unicode_errors='ignore'), encoding='utf-8')
assert_equal(re, six.u("abcdef")) assert_equal(re, six.u("abcdef"))
@raises(TypeError) @raises(TypeError)
def testNoEncoding(): def testNoEncoding():
packs(six.u("abc"), encoding=None) packb(six.u("abc"), encoding=None)
def testDecodeBinary(): def testDecodeBinary():
re = unpacks(packs("abc"), encoding=None) re = unpackb(packb("abc"), encoding=None)
assert_equal(re, b"abc") assert_equal(re, b"abc")
if __name__ == '__main__': if __name__ == '__main__':