From: Raymond Hettinger <python@rcn.com>
Date: Mon, 11 Feb 2008 20:05:53 +0000 (+0000)
Subject: Add tests for pickletools.optimize().
X-Git-Tag: v2.6a1~198
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=21290eacf6eb3ebaf01edbe4c69994aa103d608e;p=python

Add tests for pickletools.optimize().
---

diff --git a/Lib/test/test_pickletools.py b/Lib/test/test_pickletools.py
index 2e2fe50b04..3e5b35ab58 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()