From 2f501f148e8afefc88a7181d170ee1bf83327637 Mon Sep 17 00:00:00 2001 From: Thomas Kluyver <thomas.kluyver@xfel.eu> Date: Wed, 2 Aug 2023 13:02:05 +0200 Subject: [PATCH] Some fixes --- src/cal_tools/calcat_interface2.py | 4 ++-- tests/test_calcat_interface2.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/cal_tools/calcat_interface2.py b/src/cal_tools/calcat_interface2.py index ede5a89ca..456f85517 100644 --- a/src/cal_tools/calcat_interface2.py +++ b/src/cal_tools/calcat_interface2.py @@ -130,7 +130,7 @@ class CalibrationData(Mapping): cal_id_map = { api.calibration_id(calibration): calibration - for calibration in calibrations + for calibration in cal_types } calibration_ids = list(cal_id_map.keys()) @@ -153,7 +153,7 @@ class CalibrationData(Mapping): res = cls(d) if modules: - res = res.select_modules(modules) + res = res.select_modules(*modules) return res @classmethod diff --git a/tests/test_calcat_interface2.py b/tests/test_calcat_interface2.py index dc9be2ec1..9897f0a8f 100644 --- a/tests/test_calcat_interface2.py +++ b/tests/test_calcat_interface2.py @@ -20,9 +20,10 @@ def test_AGIPD_CalibrationData_metadata(): ) agipd_cd = CalibrationData.from_condition( cond, - "SPB_DET_AGIPD1M-1", + "MID_DET_AGIPD1M-1", event_at="2022-09-01 13:26:48.00", modules=['AGIPD00'], + calibrations=['Offset', 'SlopesFF'], ) assert 'Offset' in agipd_cd assert set(agipd_cd['Offset'].constants) == {'AGIPD00'} -- GitLab