diff --git a/tests/test_calcat_interface2.py b/tests/test_calcat_interface2.py index 241602eb6d0334c52c5dcc0a1c85203facae2710..6cfc68bac579e6c1c080743d591532acab49ee2c 100644 --- a/tests/test_calcat_interface2.py +++ b/tests/test_calcat_interface2.py @@ -72,6 +72,22 @@ def test_DSSC_modules_missing(): assert dssc_cd.module_nums == modnos assert dssc_cd.qm_names == [f"Q{(m // 4) + 1}M{(m % 4) + 1}" for m in modnos] + offset = dssc_cd["Offset"] + assert offset.module_nums == modnos + + # test ModulesConstantVersions.select_modules() + modnos_q3 = list(range(8, 12)) + aggs_q3 = [f"DSSC{m:02}" for m in modnos_q3] + qm_q3 = [f"Q3M{i}" for i in range(1, 5)] + assert offset.select_modules(modnos_q3).module_nums == modnos_q3 + assert offset.select_modules(aggregators=aggs_q3).module_nums == modnos_q3 + assert offset.select_modules(qm_names=qm_q3).module_nums == modnos_q3 + + # test CalibrationData.select_modules() + assert dssc_cd.select_modules(modnos_q3).module_nums == modnos_q3 + assert dssc_cd.select_modules(aggregators=aggs_q3).module_nums == modnos_q3 + assert dssc_cd.select_modules(qm_names=qm_q3).module_nums == modnos_q3 + @pytest.mark.requires_gpfs def test_LPD_constant_missing():