diff options
Diffstat (limited to 'test/test_sequnpack.py')
| -rw-r--r-- | test/test_sequnpack.py | 25 |
1 files changed, 10 insertions, 15 deletions
diff --git a/test/test_sequnpack.py b/test/test_sequnpack.py index f767726..fc1f712 100644 --- a/test/test_sequnpack.py +++ b/test/test_sequnpack.py @@ -4,7 +4,8 @@ import six from msgpack import Unpacker, BufferFull from msgpack.exceptions import OutOfData -import nose +from pytest import raises + def test_foobar(): unpacker = Unpacker(read_size=3, use_list=1) @@ -15,11 +16,8 @@ def test_foobar(): assert unpacker.unpack() == ord(b'b') assert unpacker.unpack() == ord(b'a') assert unpacker.unpack() == ord(b'r') - try: - o = unpacker.unpack() - assert 0, "should raise exception" - except OutOfData: - assert 1, "ok" + with raises(OutOfData): + unpacker.unpack() unpacker.feed(b'foo') unpacker.feed(b'bar') @@ -39,17 +37,16 @@ def test_foobar_skip(): unpacker.skip() assert unpacker.unpack() == ord(b'a') unpacker.skip() - try: - o = unpacker.unpack() - assert 0, "should raise exception" - except OutOfData: - assert 1, "ok" + with raises(OutOfData): + unpacker.unpack() def test_maxbuffersize(): - nose.tools.assert_raises(ValueError, Unpacker, read_size=5, max_buffer_size=3) + with raises(ValueError): + Unpacker(read_size=5, max_buffer_size=3) unpacker = Unpacker(read_size=3, max_buffer_size=3, use_list=1) unpacker.feed(b'fo') - nose.tools.assert_raises(BufferFull, unpacker.feed, b'ob') + with raises(BufferFull): + unpacker.feed(b'ob') unpacker.feed(b'o') assert ord('f') == next(unpacker) unpacker.feed(b'b') @@ -73,5 +70,3 @@ def test_readbytes(): assert unpacker.unpack() == ord(b'a') assert unpacker.unpack() == ord(b'r') -if __name__ == '__main__': - nose.main() |