slightly change to API

1 files changed, 4 insertions, 3 deletions

diff --git a/test/test_extension.py b/test/test_extension.py
index 0a9c14f..0b26f8e 100644
--- a/test/test_extension.py
+++ b/test/test_extension.py
@@ -3,15 +3,16 @@ import msgpack
 
 def test_extension_type():
     class MyPacker(msgpack.Packer):
-        def handle_extended_type(self, obj):
+        def handle_unknown_type(self, obj):
             if isinstance(obj, array.array):
                 fmt = "ext 32"
                 typecode = 123 # application specific typecode
                 data = obj.tostring()
-                return fmt, typecode, data
+                self.pack_extended_type(fmt, typecode, data)
+                return True
 
     class MyUnpacker(msgpack.Unpacker):
-        def handle_extended_type(self, typecode, data):
+        def read_extended_type(self, typecode, data):
             assert typecode == 123
             obj = array.array('d')
             obj.fromstring(data)