freesurfer-volume-reader/tests/conftest.py

# pylint: disable=missing-module-docstring

import os

import pandas

SUBJECTS_DIR = os.path.join(os.path.dirname(__file__), "subjects")


def assert_volume_frames_equal(left: pandas.DataFrame, right: pandas.DataFrame):
    sort_by = list(
        filter(
            lambda n: n in left, ["subject", "volume_mm^3", "analysis_id", "correction"]
        )
    )
    left.sort_values(sort_by, inplace=True)
    right.sort_values(sort_by, inplace=True)
    left.reset_index(inplace=True, drop=True)
    right.reset_index(inplace=True, drop=True)
    pandas.util.testing.assert_frame_equal(
        left=left,
        right=right,
        # ignore the order of index & columns
        check_like=True,
    )