Source code for niworkflows.interfaces.conftest

from pathlib import Path
from shutil import copytree

import pytest

try:
    from contextlib import chdir as _chdir
except ImportError:  # PY310
    import os
    from contextlib import contextmanager

    @contextmanager  # type: ignore
    def _chdir(path):
        cwd = os.getcwd()
        os.chdir(path)
        try:
            yield
        finally:
            os.chdir(cwd)




@pytest.fixture(scope='module')
def data_dir():
    return Path(__file__).parent / 'tests' / 'data'



@pytest.fixture(autouse=True)
def _docdir(request, tmp_path):
    # Trigger ONLY for the doctests.
    doctest_plugin = request.config.pluginmanager.getplugin('doctest')
    if isinstance(request.node, doctest_plugin.DoctestItem):
        copytree(Path(__file__).parent / 'tests' / 'data', tmp_path, dirs_exist_ok=True)

        # Chdir only for the duration of the test.
        with _chdir(tmp_path):
            yield

    else:
        # For normal tests, we have to yield, since this is a yield-fixture.
        yield