Initial commit of OpenSPARC T2 design and verification files.

[OpenSPARC-T2-DV] / tools / src / nas,5.n2.os.2 / lib / python / lib / python2.4 / test / test_cpickle.py

import cPickle
import unittest
from cStringIO import StringIO
from pickletester import AbstractPickleTests, AbstractPickleModuleTests
from test import test_support

class cPickleTests(AbstractPickleTests, AbstractPickleModuleTests):

   def setUp(self):
       self.dumps = cPickle.dumps
       self.loads = cPickle.loads

   error = cPickle.BadPickleGet
   module = cPickle

class cPicklePicklerTests(AbstractPickleTests):

   def dumps(self, arg, proto=0):
       f = StringIO()
       p = cPickle.Pickler(f, proto)
       p.dump(arg)
       f.seek(0)
       return f.read()

   def loads(self, buf):
       f = StringIO(buf)
       p = cPickle.Unpickler(f)
       return p.load()

   error = cPickle.BadPickleGet

class cPickleListPicklerTests(AbstractPickleTests):

   def dumps(self, arg, proto=0):
       p = cPickle.Pickler(proto)
       p.dump(arg)
       return p.getvalue()

   def loads(self, *args):
       f = StringIO(args[0])
       p = cPickle.Unpickler(f)
       return p.load()

   error = cPickle.BadPickleGet

class cPickleFastPicklerTests(AbstractPickleTests):

   def dumps(self, arg, proto=0):
       f = StringIO()
       p = cPickle.Pickler(f, proto)
       p.fast = 1
       p.dump(arg)
       f.seek(0)
       return f.read()

   def loads(self, *args):
       f = StringIO(args[0])
       p = cPickle.Unpickler(f)
       return p.load()

   error = cPickle.BadPickleGet

   def test_recursive_list(self):
       self.assertRaises(ValueError,
                         AbstractPickleTests.test_recursive_list,
                         self)

   def test_recursive_inst(self):
       self.assertRaises(ValueError,
                         AbstractPickleTests.test_recursive_inst,
                         self)

   def test_recursive_dict(self):
       self.assertRaises(ValueError,
                         AbstractPickleTests.test_recursive_dict,
                         self)

   def test_recursive_multi(self):
       self.assertRaises(ValueError,
                         AbstractPickleTests.test_recursive_multi,
                         self)

   def test_nonrecursive_deep(self):
       # If it's not cyclic, it should pickle OK even if the nesting
       # depth exceeds PY_CPICKLE_FAST_LIMIT.  That happens to be
       # 50 today.  Jack Jansen reported stack overflow on Mac OS 9
       # at 64.
       a = []
       for i in range(60):
           a = [a]
       b = self.loads(self.dumps(a))
       self.assertEqual(a, b)

def test_main():
   test_support.run_unittest(
       cPickleTests,
       cPicklePicklerTests,
       cPickleListPicklerTests,
       cPickleFastPicklerTests
   )

if __name__ == "__main__":
   test_main()