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

blacken test

Browse source
This commit is contained in:
Inada Naoki 2019-12-05 18:51:45 +09:00
parent e557e17cbd
commit 10e5e39ff9
16 changed files with 501 additions and 334 deletions
Download patch file Download diff file Expand all files Collapse all files

49
test/test_read_size.py
View file

@ -1,66 +1,71 @@
"""Test Unpacker's read_array_header and read_map_header methods"""
from msgpack import packb, Unpacker, OutOfData
UnexpectedTypeException = ValueError
def test_read_array_header():
unpacker = Unpacker()
unpacker.feed(packb(['a', 'b', 'c']))
unpacker.feed(packb(["a", "b", "c"]))
assert unpacker.read_array_header() == 3
assert unpacker.unpack() == b'a'
assert unpacker.unpack() == b'b'
assert unpacker.unpack() == b'c'
assert unpacker.unpack() == b"a"
assert unpacker.unpack() == b"b"
assert unpacker.unpack() == b"c"
try:
unpacker.unpack()
assert 0, 'should raise exception'
assert 0, "should raise exception"
except OutOfData:
assert 1, 'okay'
assert 1, "okay"
def test_read_map_header():
unpacker = Unpacker()
unpacker.feed(packb({'a': 'A'}))
unpacker.feed(packb({"a": "A"}))
assert unpacker.read_map_header() == 1
assert unpacker.unpack() == B'a'
assert unpacker.unpack() == B'A'
assert unpacker.unpack() == b"a"
assert unpacker.unpack() == b"A"
try:
unpacker.unpack()
assert 0, 'should raise exception'
assert 0, "should raise exception"
except OutOfData:
assert 1, 'okay'
assert 1, "okay"
def test_incorrect_type_array():
unpacker = Unpacker()
unpacker.feed(packb(1))
try:
unpacker.read_array_header()
assert 0, 'should raise exception'
assert 0, "should raise exception"
except UnexpectedTypeException:
assert 1, 'okay'
assert 1, "okay"
def test_incorrect_type_map():
unpacker = Unpacker()
unpacker.feed(packb(1))
try:
unpacker.read_map_header()
assert 0, 'should raise exception'
assert 0, "should raise exception"
except UnexpectedTypeException:
assert 1, 'okay'
assert 1, "okay"
def test_correct_type_nested_array():
unpacker = Unpacker()
unpacker.feed(packb({'a': ['b', 'c', 'd']}))
unpacker.feed(packb({"a": ["b", "c", "d"]}))
try:
unpacker.read_array_header()
assert 0, 'should raise exception'
assert 0, "should raise exception"
except UnexpectedTypeException:
assert 1, 'okay'
assert 1, "okay"
def test_incorrect_type_nested_map():
unpacker = Unpacker()
unpacker.feed(packb([{'a': 'b'}]))
unpacker.feed(packb([{"a": "b"}]))
try:
unpacker.read_map_header()
assert 0, 'should raise exception'
assert 0, "should raise exception"
except UnexpectedTypeException:
assert 1, 'okay'
assert 1, "okay"