summaryrefslogtreecommitdiff
path: root/test3/test_obj.py
diff options
context:
space:
mode:
Diffstat (limited to 'test3/test_obj.py')
-rw-r--r--test3/test_obj.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/test3/test_obj.py b/test3/test_obj.py
index 972d1ca..236988d 100644
--- a/test3/test_obj.py
+++ b/test3/test_obj.py
@@ -26,6 +26,19 @@ def test_decode_hook():
unpacked = unpacks(packed, object_hook=_decode_complex)
eq_(unpacked[1], 1+2j)
+@raises(TypeError)
+def test_bad_hook():
+ packed = packs([3, 1+2j], default=lambda o: o)
+ unpacked = unpacks(packed)
+
+def _arr_to_str(arr):
+ return ''.join(str(c) for c in arr)
+
+def test_array_hook():
+ packed = packs([1,2,3])
+ unpacked = unpacks(packed, list_hook=_arr_to_str)
+ eq_(unpacked, '123')
+
if __name__ == '__main__':
#main()
test_decode_hook()