| Christian Heimes | 126d29a | 2008-02-11 22:57:17 +0000 | [diff] [blame] | 1 | import pickle |
| Tim Peters | 8ecfc8e | 2003-01-27 18:51:48 +0000 | [diff] [blame] | 2 | import pickletools |
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 3 | from test import support |
| Christian Heimes | 126d29a | 2008-02-11 22:57:17 +0000 | [diff] [blame] | 4 | from test.pickletester import AbstractPickleTests |
| 5 | from test.pickletester import AbstractPickleModuleTests |
| 6 | |
| 7 | class OptimizedPickleTests(AbstractPickleTests, AbstractPickleModuleTests): |
| 8 | |
| Guido van Rossum | f416981 | 2008-03-17 22:56:06 +0000 | [diff] [blame] | 9 | def dumps(self, arg, proto=None): |
| Christian Heimes | 126d29a | 2008-02-11 22:57:17 +0000 | [diff] [blame] | 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(): |
| Benjamin Peterson | ee8712c | 2008-05-20 21:35:26 +0000 | [diff] [blame] | 19 | support.run_unittest(OptimizedPickleTests) |
| 20 | support.run_doctest(pickletools) |
| Christian Heimes | 126d29a | 2008-02-11 22:57:17 +0000 | [diff] [blame] | 21 | |
| 22 | |
| 23 | if __name__ == "__main__": |
| 24 | test_main() |