mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
865 B
Python
39 lines
865 B
Python
#!/usr/bin/env python
|
|
#
|
|
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
|
|
# Copyright (c) 2008-2016 California Institute of Technology.
|
|
# Copyright (c) 2016-2019 The Uncertainty Quantification Foundation.
|
|
# License: 3-clause BSD. The full license text is available at:
|
|
# - https://github.com/uqfoundation/dill/blob/master/LICENSE
|
|
|
|
import dill as pickle
|
|
try:
|
|
from StringIO import StringIO
|
|
except ImportError:
|
|
from io import BytesIO as StringIO
|
|
|
|
|
|
def my_fn(x):
|
|
return x * 17
|
|
|
|
|
|
def test_extend():
|
|
obj = lambda : my_fn(34)
|
|
assert obj() == 578
|
|
|
|
obj_io = StringIO()
|
|
pickler = pickle.Pickler(obj_io)
|
|
pickler.dump(obj)
|
|
|
|
obj_str = obj_io.getvalue()
|
|
|
|
obj2_io = StringIO(obj_str)
|
|
unpickler = pickle.Unpickler(obj2_io)
|
|
obj2 = unpickler.load()
|
|
|
|
assert obj2() == 578
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_extend()
|