Lib/test/test_pickle.py

   1 import pickle
   2 import test_support
   3 import unittest
   4 from cStringIO import StringIO
   5 from pickletester import AbstractPickleTests, AbstractPickleModuleTests
   6
   7 class PickleTests(AbstractPickleTests, AbstractPickleModuleTests):
   8
   9     def setUp(self):
  10         self.dumps = pickle.dumps
  11         self.loads = pickle.loads
  12
  13     module = pickle
  14     error = KeyError
  15
  16 class PicklerTests(AbstractPickleTests):
  17
  18     error = KeyError
  19
  20     def dumps(self, arg, bin=0):
  21         f = StringIO()
  22         p = pickle.Pickler(f, bin)
  23         p.dump(arg)
  24         f.seek(0)
  25         return f.read()
  26
  27     def loads(self, buf):
  28         f = StringIO(buf)
  29         u = pickle.Unpickler(f)
  30         return u.load()
  31
  32 def test_main():
  33     loader = unittest.TestLoader()
  34     suite = unittest.TestSuite()
  35     suite.addTest(loader.loadTestsFromTestCase(PickleTests))
  36     suite.addTest(loader.loadTestsFromTestCase(PicklerTests))
  37     test_support.run_suite(suite)
  38
  39 if __name__ == "__main__":
  40     test_main()