mirror of
https://github.com/python/cpython.git
synced 2025-07-23 11:15:24 +00:00
7107a7fbcc
test_nonrecursive_deep(): Reduced nesting depth to 60. Not a bugfix candidate. 2.3 increased the number of stack frames needed to pickle a list (in order to get implement the "list batching" unpickling memory optimization new in 2.3).
104 lines
2.9 KiB
Python
104 lines
2.9 KiB
Python
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():
|
|
loader = unittest.TestLoader()
|
|
suite = unittest.TestSuite()
|
|
suite.addTest(loader.loadTestsFromTestCase(cPickleTests))
|
|
suite.addTest(loader.loadTestsFromTestCase(cPicklePicklerTests))
|
|
suite.addTest(loader.loadTestsFromTestCase(cPickleListPicklerTests))
|
|
suite.addTest(loader.loadTestsFromTestCase(cPickleFastPicklerTests))
|
|
test_support.run_suite(suite)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|