#!/usr/bin/env python # # Author: Mike McKerns (mmckerns @caltech and @uqfoundation) # Copyright (c) 2008-2016 California Institute of Technology. # Copyright (c) 2016-2023 The Uncertainty Quantification Foundation. # License: 3-clause BSD. The full license text is available at: # - https://github.com/uqfoundation/dill/blob/master/LICENSE """ demonstrate dill's ability to pickle different python types test pickling of all Python Standard Library objects (currently: CH 1-14 @ 2.7) """ import dill as pickle pickle.settings['recurse'] = True #pickle.detect.trace(True) #import pickle # get all objects for testing from dill import load_types, objects, extend load_types(pickleable=True,unpickleable=False) # uncomment the next two lines to test cloudpickle #extend(False) #import cloudpickle as pickle # helper objects class _class: def _method(self): pass # objects that *fail* if imported special = {} special['LambdaType'] = _lambda = lambda x: lambda y: x special['MethodType'] = _method = _class()._method special['UnboundMethodType'] = _class._method objects.update(special) def pickles(name, exact=False): """quick check if object pickles with dill""" obj = objects[name] try: pik = pickle.loads(pickle.dumps(obj)) if exact: try: assert pik == obj except AssertionError: assert type(obj) == type(pik) print ("weak: %s %s" % (name, type(obj))) else: assert type(obj) == type(pik) except Exception: print ("fails: %s %s" % (name, type(obj))) def test_objects(): for member in objects.keys(): #pickles(member, exact=True) pickles(member, exact=False) if __name__ == '__main__': import warnings warnings.simplefilter('ignore') test_objects()