1 import pickle

     2 import pickletools

     3 from test import test_support

     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()