python: Port some tests from 2 to 3.

1 files changed, 13 insertions, 0 deletions

diff --git a/python/test3/test_obj.py b/python/test3/test_obj.py
index 972d1ca..236988d 100644
--- a/python/test3/test_obj.py
+++ b/python/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()