[Python-checkins] r60734 - python/trunk/Lib/test/test_pickletools.py

Mon Feb 11 21:05:53 CET 2008

Author: raymond.hettinger
Date: Mon Feb 11 21:05:53 2008
New Revision: 60734
Modified:
 python/trunk/Lib/test/test_pickletools.py
Log:
Add tests for pickletools.optimize().
Modified: python/trunk/Lib/test/test_pickletools.py
==============================================================================

--- python/trunk/Lib/test/test_pickletools.py	(original)
+++ python/trunk/Lib/test/test_pickletools.py	Mon Feb 11 21:05:53 2008
@@ -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()
