Function tempdir

numpy/testing/_private/utils.py:1902–1913

Context manager to provide a temporary test folder. All arguments are passed as this to the underlying tempfile.mkdtemp function.

(*args, **kwargs)

@contextlib.contextmanager
def tempdir(*args, **kwargs):
    """Context manager to provide a temporary test folder.

    All arguments are passed as this to the underlying tempfile.mkdtemp
    function.

    """
    tmpdir = mkdtemp(*args, **kwargs)
    try:
        yield tmpdir
    finally:
        shutil.rmtree(tmpdir)


@contextlib.contextmanager