123456789101112131415161718192021222324252627282930313233343536373839 |
- # coding: utf-8
- from datetime import datetime
- import pytest
- from pandas.io.msgpack import packb, unpackb
- class DummyException(Exception):
- pass
- class TestExceptions(object):
- def test_raise_on_find_unsupported_value(self):
- msg = "can\'t serialize datetime"
- with pytest.raises(TypeError, match=msg):
- packb(datetime.now())
- def test_raise_from_object_hook(self):
- def hook(_):
- raise DummyException()
- with pytest.raises(DummyException):
- unpackb(packb({}), object_hook=hook)
- with pytest.raises(DummyException):
- unpackb(packb({'fizz': 'buzz'}), object_hook=hook)
- with pytest.raises(DummyException):
- unpackb(packb({'fizz': 'buzz'}), object_pairs_hook=hook)
- with pytest.raises(DummyException):
- unpackb(packb({'fizz': {'buzz': 'spam'}}), object_hook=hook)
- with pytest.raises(DummyException):
- unpackb(packb({'fizz': {'buzz': 'spam'}}), object_pairs_hook=hook)
- def test_invalid_value(self):
- msg = "Unpack failed: error"
- with pytest.raises(ValueError, match=msg):
- unpackb(b"\xd9\x97#DL_")
|