1 # test_pickle dumps and loads pickles via pickle.py.

     2 # test_cpickle does the same, but via the cPickle module.

     3 # This test covers the other two cases, making pickles with one module and

     4 # loading them via the other.

     5 

     6 import pickle

     7 import cPickle

     8 

     9 from test import test_support

    10 from test.pickletester import AbstractPickleTests

    11 

    12 class DumpCPickle_LoadPickle(AbstractPickleTests):

    13 

    14     error = KeyError

    15 

    16     def dumps(self, arg, proto=0, fast=0):

    17         # Ignore fast

    18         return cPickle.dumps(arg, proto)

    19 

    20     def loads(self, buf):

    21         # Ignore fast

    22         return pickle.loads(buf)

    23 

    24 class DumpPickle_LoadCPickle(AbstractPickleTests):

    25 

    26     error = cPickle.BadPickleGet

    27 

    28     def dumps(self, arg, proto=0, fast=0):

    29         # Ignore fast

    30         return pickle.dumps(arg, proto)

    31 

    32     def loads(self, buf):

    33         # Ignore fast

    34         return cPickle.loads(buf)

    35 

    36 def test_main():

    37     test_support.run_unittest(

    38         DumpCPickle_LoadPickle,

    39         DumpPickle_LoadCPickle

    40     )

    41 

    42 if __name__ == "__main__":

    43     test_main()