mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-10-23 13:53:18 +00:00
parent
7a8ce0f9ca
commit
f6f6f328eb
2 changed files with 12 additions and 1 deletions
|
@ -357,7 +357,9 @@ class Unpacker(object):
|
||||||
return self._buffer[self._buff_i :]
|
return self._buffer[self._buff_i :]
|
||||||
|
|
||||||
def read_bytes(self, n):
|
def read_bytes(self, n):
|
||||||
return self._read(n)
|
ret = self._read(n)
|
||||||
|
self._consume()
|
||||||
|
return ret
|
||||||
|
|
||||||
def _read(self, n):
|
def _read(self, n):
|
||||||
# (int) -> bytearray
|
# (int) -> bytearray
|
||||||
|
|
|
@ -93,6 +93,15 @@ def test_readbytes():
|
||||||
assert unpacker.unpack() == ord(b"a")
|
assert unpacker.unpack() == ord(b"a")
|
||||||
assert unpacker.unpack() == ord(b"r")
|
assert unpacker.unpack() == ord(b"r")
|
||||||
|
|
||||||
|
# Issue 352
|
||||||
|
u = Unpacker()
|
||||||
|
u.feed(b"x")
|
||||||
|
assert bytes(u.read_bytes(1)) == b"x"
|
||||||
|
with raises(StopIteration):
|
||||||
|
next(u)
|
||||||
|
u.feed(b"\1")
|
||||||
|
assert next(u) == 1
|
||||||
|
|
||||||
|
|
||||||
def test_issue124():
|
def test_issue124():
|
||||||
unpacker = Unpacker()
|
unpacker = Unpacker()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue