1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
| import pickle
| import pickletools
| from test import test_support
| 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()
|
|
