home *** CD-ROM | disk | FTP | other *** search
/ H4CK3R 14 / hacker14.iso / programacao / pythonwin / python.exe / TEST_CPICKLE.PY < prev    next >
Encoding:
Python Source  |  2003-05-01  |  2.8 KB  |  104 lines
  1. import cPickle
  2. import unittest
  3. from cStringIO import StringIO
  4. from pickletester import AbstractPickleTests, AbstractPickleModuleTests
  5. from test import test_support
  6.  
  7. class cPickleTests(AbstractPickleTests, AbstractPickleModuleTests):
  8.  
  9.     def setUp(self):
  10.         self.dumps = cPickle.dumps
  11.         self.loads = cPickle.loads
  12.  
  13.     error = cPickle.BadPickleGet
  14.     module = cPickle
  15.  
  16. class cPicklePicklerTests(AbstractPickleTests):
  17.  
  18.     def dumps(self, arg, proto=0):
  19.         f = StringIO()
  20.         p = cPickle.Pickler(f, proto)
  21.         p.dump(arg)
  22.         f.seek(0)
  23.         return f.read()
  24.  
  25.     def loads(self, buf):
  26.         f = StringIO(buf)
  27.         p = cPickle.Unpickler(f)
  28.         return p.load()
  29.  
  30.     error = cPickle.BadPickleGet
  31.  
  32. class cPickleListPicklerTests(AbstractPickleTests):
  33.  
  34.     def dumps(self, arg, proto=0):
  35.         p = cPickle.Pickler(proto)
  36.         p.dump(arg)
  37.         return p.getvalue()
  38.  
  39.     def loads(self, *args):
  40.         f = StringIO(args[0])
  41.         p = cPickle.Unpickler(f)
  42.         return p.load()
  43.  
  44.     error = cPickle.BadPickleGet
  45.  
  46. class cPickleFastPicklerTests(AbstractPickleTests):
  47.  
  48.     def dumps(self, arg, proto=0):
  49.         f = StringIO()
  50.         p = cPickle.Pickler(f, proto)
  51.         p.fast = 1
  52.         p.dump(arg)
  53.         f.seek(0)
  54.         return f.read()
  55.  
  56.     def loads(self, *args):
  57.         f = StringIO(args[0])
  58.         p = cPickle.Unpickler(f)
  59.         return p.load()
  60.  
  61.     error = cPickle.BadPickleGet
  62.  
  63.     def test_recursive_list(self):
  64.         self.assertRaises(ValueError,
  65.                           AbstractPickleTests.test_recursive_list,
  66.                           self)
  67.  
  68.     def test_recursive_inst(self):
  69.         self.assertRaises(ValueError,
  70.                           AbstractPickleTests.test_recursive_inst,
  71.                           self)
  72.  
  73.     def test_recursive_dict(self):
  74.         self.assertRaises(ValueError,
  75.                           AbstractPickleTests.test_recursive_dict,
  76.                           self)
  77.  
  78.     def test_recursive_multi(self):
  79.         self.assertRaises(ValueError,
  80.                           AbstractPickleTests.test_recursive_multi,
  81.                           self)
  82.  
  83.     def test_nonrecursive_deep(self):
  84.         # If it's not cyclic, it should pickle OK even if the nesting
  85.         # depth exceeds PY_CPICKLE_FAST_LIMIT.  That happens to be
  86.         # 50 today.  Jack Jansen reported stack overflow on Mac OS 9
  87.         # at 64.
  88.         a = []
  89.         for i in range(60):
  90.             a = [a]
  91.         b = self.loads(self.dumps(a))
  92.         self.assertEqual(a, b)
  93.  
  94. def test_main():
  95.     test_support.run_unittest(
  96.         cPickleTests,
  97.         cPicklePicklerTests,
  98.         cPickleListPicklerTests,
  99.         cPickleFastPicklerTests
  100.     )
  101.  
  102. if __name__ == "__main__":
  103.     test_main()
  104.