#!/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 import sys from dill.temp import dump, dump_source, dumpIO, dumpIO_source from dill.temp import load, load_source, loadIO, loadIO_source WINDOWS = sys.platform[:3] == 'win' f = lambda x: x**2 x = [1,2,3,4,5] # source code to tempfile def test_code_to_tempfile(): if not WINDOWS: #see: https://bugs.python.org/issue14243 pyfile = dump_source(f, alias='_f') _f = load_source(pyfile) assert _f(4) == f(4) # source code to stream def test_code_to_stream(): pyfile = dumpIO_source(f, alias='_f') _f = loadIO_source(pyfile) assert _f(4) == f(4) # pickle to tempfile def test_pickle_to_tempfile(): if not WINDOWS: #see: https://bugs.python.org/issue14243 dumpfile = dump(x) _x = load(dumpfile) assert _x == x # pickle to stream def test_pickle_to_stream(): dumpfile = dumpIO(x) _x = loadIO(dumpfile) assert _x == x ### now testing the objects ### f = lambda x: x**2 def g(x): return f(x) - x def h(x): def g(x): return x return g(x) - x class Foo(object): def bar(self, x): return x*x+x _foo = Foo() def add(x,y): return x+y # yes, same as 'f', but things are tricky when it comes to pointers squared = lambda x:x**2 class Bar: pass _bar = Bar() # test function-type objects that take 2 args def test_two_arg_functions(): for obj in [add]: pyfile = dumpIO_source(obj, alias='_obj') _obj = loadIO_source(pyfile) assert _obj(4,2) == obj(4,2) # test function-type objects that take 1 arg def test_one_arg_functions(): for obj in [g, h, squared]: pyfile = dumpIO_source(obj, alias='_obj') _obj = loadIO_source(pyfile) assert _obj(4) == obj(4) # test instance-type objects #for obj in [_bar, _foo]: # pyfile = dumpIO_source(obj, alias='_obj') # _obj = loadIO_source(pyfile) # assert type(_obj) == type(obj) # test the rest of the objects def test_the_rest(): for obj in [Bar, Foo, Foo.bar, _foo.bar]: pyfile = dumpIO_source(obj, alias='_obj') _obj = loadIO_source(pyfile) assert _obj.__name__ == obj.__name__ if __name__ == '__main__': test_code_to_tempfile() test_code_to_stream() test_pickle_to_tempfile() test_pickle_to_stream() test_two_arg_functions() test_one_arg_functions() test_the_rest()