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