| Raymond Hettinger | 21290ea | 2008-02-11 20:05:53 +0000 | [diff] [blame] | 1 | import pickle |
| Tim Peters | 8ecfc8e | 2003-01-27 18:51:48 +0000 | [diff] [blame] | 2 | import pickletools |
| 3 | from test import test_support |
| Raymond Hettinger | 21290ea | 2008-02-11 20:05:53 +0000 | [diff] [blame] | 4 | from test.pickletester import AbstractPickleTests |
| 5 | from test.pickletester import AbstractPickleModuleTests |
| 6 | |
| 7 | class OptimizedPickleTests(AbstractPickleTests, AbstractPickleModuleTests): |
| 8 | |
| 9 | def dumps(self, arg, proto=0, fast=0): |
| 10 | return pickletools.optimize(pickle.dumps(arg, proto)) |
| 11 | |
| 12 | def loads(self, buf): |
| 13 | return pickle.loads(buf) |
| 14 | |
| 15 | module = pickle |
| 16 | error = KeyError |
| 17 | |
| 18 | def test_main(): |
| 19 | test_support.run_unittest(OptimizedPickleTests) |
| 20 | test_support.run_doctest(pickletools) |
| 21 | |
| 22 | |
| 23 | if __name__ == "__main__": |
| 24 | test_main() |