# coding: utf-8
from datetime import datetime
import pytest
from pandas.io.msgpack import packb, unpackb
class DummyException(Exception):
pass
class TestExceptions:
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_")