56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
#
|
||
|
|
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
|
||
|
|
# Copyright (c) 2021-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 dill
|
||
|
|
dill.settings['recurse'] = True
|
||
|
|
|
||
|
|
def get_fun_with_strftime():
|
||
|
|
def fun_with_strftime():
|
||
|
|
import datetime
|
||
|
|
return datetime.datetime.strptime("04-01-1943", "%d-%m-%Y").strftime(
|
||
|
|
"%Y-%m-%d %H:%M:%S"
|
||
|
|
)
|
||
|
|
return fun_with_strftime
|
||
|
|
|
||
|
|
|
||
|
|
def get_fun_with_strftime2():
|
||
|
|
import datetime
|
||
|
|
return datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||
|
|
|
||
|
|
|
||
|
|
def test_doc_dill_issue_219():
|
||
|
|
back_fn = dill.loads(dill.dumps(get_fun_with_strftime()))
|
||
|
|
assert back_fn() == "1943-01-04 00:00:00"
|
||
|
|
dupl = dill.loads(dill.dumps(get_fun_with_strftime2))
|
||
|
|
assert dupl() == get_fun_with_strftime2()
|
||
|
|
|
||
|
|
|
||
|
|
def get_fun_with_internal_import():
|
||
|
|
def fun_with_import():
|
||
|
|
import re
|
||
|
|
return re.compile("$")
|
||
|
|
return fun_with_import
|
||
|
|
|
||
|
|
|
||
|
|
def test_method_with_internal_import_should_work():
|
||
|
|
import re
|
||
|
|
back_fn = dill.loads(dill.dumps(get_fun_with_internal_import()))
|
||
|
|
import inspect
|
||
|
|
if hasattr(inspect, 'getclosurevars'):
|
||
|
|
vars = inspect.getclosurevars(back_fn)
|
||
|
|
assert vars.globals == {}
|
||
|
|
assert vars.nonlocals == {}
|
||
|
|
assert back_fn() == re.compile("$")
|
||
|
|
assert "__builtins__" in back_fn.__globals__
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import sys
|
||
|
|
if (sys.version_info[:3] != (3,10,0) or sys.version_info[3] != 'alpha'):
|
||
|
|
test_doc_dill_issue_219()
|
||
|
|
test_method_with_internal_import_should_work()
|