import pytest
from dvc.repo import Repo
from loguru import logger

from cv_analysis.locations import REPO_ROOT_PATH, TEST_DATA_DIR_DVC


@pytest.fixture(scope="session")
def dvc_test_data():
    # noinspection PyCallingNonCallable
    logger.info("Pulling data with DVC...")
    Repo(REPO_ROOT_PATH).pull(targets=[str(TEST_DATA_DIR_DVC)])
    logger.info("Finished pulling data.")