1
1
# coding: utf-8
2
2
3
- import pytest
4
-
3
+ from datetime import datetime
5
4
from pandas .io .msgpack import packb , unpackb
6
5
6
+ import pytest
7
+ import pandas .util .testing as tm
8
+
7
9
8
10
class DummyException (Exception ):
9
11
pass
@@ -12,12 +14,13 @@ class DummyException(Exception):
12
14
class TestExceptions (object ):
13
15
14
16
def test_raise_on_find_unsupported_value (self ):
15
- import datetime
16
- pytest .raises (TypeError , packb , datetime .datetime .now ())
17
+ msg = "can\' t serialize datetime"
18
+ with tm .assert_raises_regex (TypeError , msg ):
19
+ packb (datetime .now ())
17
20
18
21
def test_raise_from_object_hook (self ):
19
- def hook (obj ):
20
- raise DummyException
22
+ def hook (_ ):
23
+ raise DummyException ()
21
24
22
25
pytest .raises (DummyException , unpackb , packb ({}), object_hook = hook )
23
26
pytest .raises (DummyException , unpackb , packb ({'fizz' : 'buzz' }),
@@ -30,5 +33,7 @@ def hook(obj):
30
33
packb ({'fizz' : {'buzz' : 'spam' }}),
31
34
object_pairs_hook = hook )
32
35
33
- def test_invalidvalue (self ):
34
- pytest .raises (ValueError , unpackb , b'\xd9 \x97 #DL_' )
36
+ def test_invalid_value (self ):
37
+ msg = "Unpack failed: error"
38
+ with tm .assert_raises_regex (ValueError , msg ):
39
+ unpackb (b"\xd9 \x97 #DL_" )
0 commit comments