From 21290eacf6eb3ebaf01edbe4c69994aa103d608e Mon Sep 17 00:00:00 2001 From: Raymond Hettinger Date: Mon, 11 Feb 2008 20:05:53 +0000 Subject: Add tests for pickletools.optimize(). --- Lib/test/test_pickletools.py | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/Lib/test/test_pickletools.py b/Lib/test/test_pickletools.py index 2e2fe50..3e5b35a 100644 --- a/Lib/test/test_pickletools.py +++ b/Lib/test/test_pickletools.py @@ -1,3 +1,24 @@ +import pickle import pickletools from test import test_support -test_support.run_doctest(pickletools) +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() -- cgit v0.12