conftest.py 607 B

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