home *** CD-ROM | disk | FTP | other *** search
- import pickle
- import pickletools
- from test import test_support
- from test.pickletester import AbstractPickleTests
- from test.pickletester import AbstractPickleModuleTests
-
- class OptimizedPickleTests(AbstractPickleTests, AbstractPickleModuleTests):
-
- def dumps(self, arg, proto=0, fast=0):
- return pickletools.optimize(pickle.dumps(arg, proto))
-
- def loads(self, buf):
- return pickle.loads(buf)
-
- module = pickle
- error = KeyError
-
- def test_main():
- test_support.run_unittest(OptimizedPickleTests)
- test_support.run_doctest(pickletools)
-
-
- if __name__ == "__main__":
- test_main()
-