#!/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()