manyi/venv/Lib/site-packages/pytest_datadir/plugin.py

import os
import pathlib
import shutil

import pytest


@pytest.fixture
def shared_datadir(request, tmpdir):
    original_shared_path = os.path.join(request.fspath.dirname, 'data')
    temp_path = pathlib.Path(str(tmpdir.join('data')))
    shutil.copytree(original_shared_path, str(temp_path))
    return temp_path


@pytest.fixture
def original_datadir(request):
    return pathlib.Path(os.path.splitext(request.module.__file__)[0])


@pytest.fixture
def datadir(original_datadir, tmpdir):
    result = pathlib.Path(str(tmpdir.join(original_datadir.stem)))
    shutil.copytree(str(original_datadir), str(result))
    return result