import os

import pandas as pd
import pytest

from user_similarity_model.config.core import DATASET_DIR, config


@pytest.fixture()
def sample_local_data():
    """AI is creating summary for sample_local_data

    Returns:
        [Dict]: This function returns a dictionary with CSV files which
        in dataset folder. The data will be compared in tests against data
        that are pulled from Azure PostgreSQL server.
    """
    sample_data = {}
    for file in config.app_config.csv_files:
        sample_data[file[0:-4]] = pd.read_csv(os.path.join(DATASET_DIR, file))
    return sample_data

[((9, 1, 9, 17), 'pytest.fixture', 'pytest.fixture', ({}, {}), '()', False, 'import pytest\n'), ((20, 46, 20, 77), 'os.path.join', 'os.path.join', ({(20, 59, 20, 70): 'DATASET_DIR', (20, 72, 20, 76): 'file'}, {}), '(DATASET_DIR, file)', False, 'import os\n')]