diff options
author | INADA Naoki <inada-n@klab.com> | 2014-02-13 03:10:51 +0900 |
---|---|---|
committer | INADA Naoki <inada-n@klab.com> | 2014-02-13 03:10:51 +0900 |
commit | 9d61f243878eeabd2042bb16fe22d4325e441da6 (patch) | |
tree | 2364417fcb3e9f6aa0d5491f85b6e47aff979d7c /test | |
parent | d2fc8010342512378e01322f8871c10a5974af4f (diff) | |
download | msgpack-python-9d61f243878eeabd2042bb16fe22d4325e441da6.tar.gz |
Feed data from file before _unpack()
Diffstat (limited to 'test')
-rw-r--r-- | test/test_unpack_file.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/test/test_unpack_file.py b/test/test_unpack_file.py new file mode 100644 index 0000000..1563008 --- /dev/null +++ b/test/test_unpack_file.py @@ -0,0 +1,19 @@ +from io import BytesIO +from msgpack import Unpacker, packb, OutOfData +from pytest import raises + + +def test_unpack_array_header_from_file(): + f = BytesIO(packb([1,2,3,4])) + unpacker = Unpacker(f) + assert unpacker.read_array_header() == 4 + assert unpacker.unpack() == 1 + assert unpacker.unpack() == 2 + assert unpacker.unpack() == 3 + assert unpacker.unpack() == 4 + with raises(OutOfData): + unpacker.unpack() + + +if __name__ == '__main__': + test_unpack_array_header_from_file() |