fixed file dump and load which not works

Author: ChameleonRed

cw_msgpack_coder/umsgpack_coder.py

@@ -149,15 +149,15 @@ def _ext_load(self):
 
     def dump(self, obj, f):
         """ Dump object to file stream. """
-        umsgpack.pack(obj, f, ext_handlers=self._ext_dump())
+        return umsgpack.pack(obj, f, ext_handlers=self._ext_dump())
 
     def dumps(self, obj):
         """ Dump object to bytes array. """
         return umsgpack.packb(obj, ext_handlers=self._ext_dump())
 
     def load(self, f):
         """ Load object from file stream. """
-        umsgpack.unpack(f, ext_handlers=self._ext_load())
+        return umsgpack.unpack(f, ext_handlers=self._ext_load())
 
     def loads(self, b):
         """ Load object from bytes array. """

setup.py

@@ -9,7 +9,7 @@ def long_description():
 
 
 setup(name='cw_msgpack_coder',
-      version='1.1.2',
+      version='1.1.3',
       description='Simple and fast Python any object serialization with use msgpack.',
       long_description=long_description(),
       author='Cezary K. Wagner',

tests/test_umsgpack_coder.py

@@ -1,4 +1,5 @@
 import unittest
+import io
 
 from cw_msgpack_coder.umsgpack_coder import UmsgpackCoder
 
@@ -20,6 +21,13 @@ def test_encode_empty_class(self):
         o2 = coder.loads(s)
         self.assertEqual(o, o2)
 
+        stream = io.BytesIO()
+        coder.dump(o, stream)
+        stream.seek(0)
+        o2 = coder.load(stream)
+        self.assertEqual(o, o2)
+
+
     class DictClass:
         def __init__(self, a):
             self.a = a
@@ -40,6 +48,12 @@ def test_encode_dict_class(self):
         o2 = coder.loads(s)
         self.assertEqual(o, o2)
 
+        stream = io.BytesIO()
+        coder.dump(o, stream)
+        stream.seek(0)
+        o2 = coder.load(stream)
+        self.assertEqual(o, o2)
+
     class SlotClass:
         __slots__ = 'ab'
 
@@ -60,6 +74,12 @@ def test_encode_single_slot_class(self):
         o2 = coder.loads(s)
         self.assertEqual(o, o2)
 
+        stream = io.BytesIO()
+        coder.dump(o, stream)
+        stream.seek(0)
+        o2 = coder.load(stream)
+        self.assertEqual(o, o2)
+
     class MultiSlotClass:
         __slots__ = 'a', 'b'
 
@@ -82,6 +102,12 @@ def test_encode_multi_slot_class(self):
         o2 = coder.loads(s)
         self.assertEqual(o, o2)
 
+        stream = io.BytesIO()
+        coder.dump(o, stream)
+        stream.seek(0)
+        o2 = coder.load(stream)
+        self.assertEqual(o, o2)
+
     class FirstComponentClass:
         def __init__(self, a):
             self.a = a
@@ -126,6 +152,12 @@ def test_encode_compound_class(self):
         o2 = coder.loads(s)
         self.assertEqual(o, o2)
 
+        stream = io.BytesIO()
+        coder.dump(o, stream)
+        stream.seek(0)
+        o2 = coder.load(stream)
+        self.assertEqual(o, o2)
+
     def test_encode_compound_of_compound_class(self):
         coder = UmsgpackCoder()
         coder.set_default_coder_for_class(self.CompoundClass)
@@ -138,3 +170,9 @@ def test_encode_compound_of_compound_class(self):
         o2 = coder.loads(s)
         self.assertEqual(o, o2)
 
+        stream = io.BytesIO()
+        coder.dump(o, stream)
+        stream.seek(0)
+        o2 = coder.load(stream)
+        self.assertEqual(o, o2)
+