conftest

Complexity

Total Complexity

7

Size/Duplication

Total Lines	35
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	reference_data()	0	14	4
A	data_dir()	0	3	1
A	ccdc_enabled()	0	7	2

import os
import pickle
import pytest
import pathlib


@pytest.fixture(scope='session', autouse=True)
def data_dir():
    return pathlib.Path(__file__).absolute().parent / 'data'


@pytest.fixture(scope='session', autouse=True)
def ccdc_enabled():
    try:
        import ccdc
        return True
    except Exception:
        return False


@pytest.fixture(scope='session', autouse=True)
def reference_data(data_dir):
    """Data fixture of amd.PeriodicSet objects to use in tests."""

    filenames = ['cubic', 'T2_experimental', 'CSD_families']
    refs = {}
    for name in filenames:
        path = str(data_dir / f'{name}.pkl')
        with open(path, 'rb') as f:
            data = pickle.load(f)
            if not data:
                raise ValueError(f'Data not found in path {path}')
            refs[name] = data
    return refs