diff -r 000000000000 -r ae805ac0140d python-2.5.2/win32/Lib/test/test_pickle.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/python-2.5.2/win32/Lib/test/test_pickle.py Fri Apr 03 17:19:34 2009 +0100 @@ -0,0 +1,73 @@ +import pickle +import unittest +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()