Lib/test/test_cpickle.py

   1 import cPickle
   2 import test_support
   3 import unittest
   4 from cStringIO import StringIO
   5 from pickletester import AbstractPickleTests, AbstractPickleModuleTests
   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, bin=0):
  19         f = StringIO()
  20         p = cPickle.Pickler(f, bin)
  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, bin=0):
  35         p = cPickle.Pickler(bin)
  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, bin=0):
  49         f = StringIO()
  50         p = cPickle.Pickler(f, bin)
  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         a = []
  85         for i in range(100):
  86             a = [a]
  87         b = self.loads(self.dumps(a))
  88         self.assertEqual(a, b)
  89
  90 def test_main():
  91     loader = unittest.TestLoader()
  92     suite = unittest.TestSuite()
  93     suite.addTest(loader.loadTestsFromTestCase(cPickleTests))
  94     suite.addTest(loader.loadTestsFromTestCase(cPicklePicklerTests))
  95     suite.addTest(loader.loadTestsFromTestCase(cPickleListPicklerTests))
  96     suite.addTest(loader.loadTestsFromTestCase(cPickleFastPicklerTests))
  97     test_support.run_suite(suite)
  98
  99 if __name__ == "__main__":
 100     test_main()