| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- #!/usr/bin/env python
- #
- # Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
- # Copyright (c) 2021-2024 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()
|
