diff options
Diffstat (limited to 'test/test_obj.py')
| -rw-r--r-- | test/test_obj.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/test/test_obj.py b/test/test_obj.py index 881e627..1d9024b 100644 --- a/test/test_obj.py +++ b/test/test_obj.py @@ -49,5 +49,26 @@ def test_array_hook(): unpacked = unpackb(packed, list_hook=_arr_to_str, use_list=1) eq_(unpacked, '123') + +class DecodeError(Exception): + pass + +def bad_complex_decoder(o): + raise DecodeError("Ooops!") + + +@raises(DecodeError) +def test_an_exception_in_objecthook1(): + packed = packb({1: {'__complex__': True, 'real': 1, 'imag': 2}}) + unpackb(packed, object_hook=bad_complex_decoder) + + +@raises(DecodeError) +def test_an_exception_in_objecthook2(): + packed = packb({1: [{'__complex__': True, 'real': 1, 'imag': 2}]}) + unpackb(packed, list_hook=bad_complex_decoder, use_list=1) + + + if __name__ == '__main__': main() |
