diff -r ffa851df0825 -r 2fb8b9db1c86 symbian-qemu-0.9.1-12/python-2.6.1/Lib/test/test_pickle.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/symbian-qemu-0.9.1-12/python-2.6.1/Lib/test/test_pickle.py Fri Jul 31 15:01:17 2009 +0100 @@ -0,0 +1,72 @@ +import pickle +from cStringIO import StringIO + +from test import test_support + +from test.pickletester import AbstractPickleTests +from test.pickletester import AbstractPickleModuleTests +from test.pickletester import AbstractPersistentPicklerTests + +class PickleTests(AbstractPickleTests, AbstractPickleModuleTests): + + def dumps(self, arg, proto=0, fast=0): + # Ignore fast + return pickle.dumps(arg, proto) + + def loads(self, buf): + # Ignore fast + return pickle.loads(buf) + + module = pickle + error = KeyError + +class PicklerTests(AbstractPickleTests): + + error = KeyError + + def dumps(self, arg, proto=0, fast=0): + f = StringIO() + p = pickle.Pickler(f, proto) + if fast: + p.fast = fast + p.dump(arg) + f.seek(0) + return f.read() + + def loads(self, buf): + f = StringIO(buf) + u = pickle.Unpickler(f) + return u.load() + +class PersPicklerTests(AbstractPersistentPicklerTests): + + def dumps(self, arg, proto=0, fast=0): + class PersPickler(pickle.Pickler): + def persistent_id(subself, obj): + return self.persistent_id(obj) + f = StringIO() + p = PersPickler(f, proto) + if fast: + p.fast = fast + p.dump(arg) + f.seek(0) + return f.read() + + def loads(self, buf): + class PersUnpickler(pickle.Unpickler): + def persistent_load(subself, obj): + return self.persistent_load(obj) + f = StringIO(buf) + u = PersUnpickler(f) + return u.load() + +def test_main(): + test_support.run_unittest( + PickleTests, + PicklerTests, + PersPicklerTests + ) + test_support.run_doctest(pickle) + +if __name__ == "__main__": + test_main()