Merge pull request #492 from niphlod/tests/storage_pickling

added test for storage pickling
This commit is contained in:
mdipierro
2014-09-10 12:26:26 -05:00
+8
View File
@@ -37,6 +37,7 @@ def fix_sys_path():
fix_sys_path()
from storage import Storage
import pickle
class TestStorage(unittest.TestCase):
@@ -96,6 +97,13 @@ class TestStorage(unittest.TestCase):
self.assertEquals(s.a, None)
self.assertEquals(s['a'], None)
self.assertTrue('a' in s)
def test_pickling(self):
""" Test storage pickling """
s = Storage(a=1)
sd = pickle.dumps(s, pickle.HIGHEST_PROTOCOL)
news = pickle.loads(sd)
self.assertEqual(news.a, 1)
if __name__ == '__main__':
unittest.main()