conftest

Complexity

Total Complexity

7

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	data_dir()	0	3	1
A	reference_data()	0	14	4
A	ccdc_enabled()	0	8	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