summaryrefslogtreecommitdiff
path: root/test/test_read_size.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_read_size.py')
-rw-r--r--test/test_read_size.py49
1 files changed, 27 insertions, 22 deletions
diff --git a/test/test_read_size.py b/test/test_read_size.py
index 4e6c2b9..8d8df64 100644
--- a/test/test_read_size.py
+++ b/test/test_read_size.py
@@ -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"