conftest.py 686 B

12345678910111213141516171819202122232425
  1. import os
  2. import pandas
  3. import pytest
  4. SUBJECTS_DIR = os.path.join(os.path.dirname(__file__), 'subjects')
  5. def _assert_volume_frames_equal(left: pandas.DataFrame, right: pandas.DataFrame):
  6. sort_by = ['volume_mm^3', 'analysis_id']
  7. left.sort_values(sort_by, inplace=True)
  8. right.sort_values(sort_by, inplace=True)
  9. left.reset_index(inplace=True, drop=True)
  10. right.reset_index(inplace=True, drop=True)
  11. pandas.util.testing.assert_frame_equal(
  12. left=left,
  13. right=right,
  14. # ignore the order of index & columns
  15. check_like=True,
  16. )
  17. @pytest.fixture(scope='module')
  18. def assert_volume_frames_equal():
  19. return _assert_volume_frames_equal