From 3fd77427e028ee063bec1ffc1c181b296c1fd2aa Mon Sep 17 00:00:00 2001
From: ahmedk <karim.ahmed@xfel.eu>
Date: Tue, 30 Jan 2024 18:21:16 +0100
Subject: [PATCH] add tests

---
 tests/test_lpdlib.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 tests/test_lpdlib.py

diff --git a/tests/test_lpdlib.py b/tests/test_lpdlib.py
new file mode 100644
index 000000000..5648c5c64
--- /dev/null
+++ b/tests/test_lpdlib.py
@@ -0,0 +1,27 @@
+import pytest
+
+from cal_tools.lpdlib import sort_dark_runs_by_gain
+
+
+@pytest.mark.requires_gpfs
+def test_sort_dark_runs_by_gain():
+    # TODO: update used raw data to test data when available.
+    raw_folder = "/gpfs/exfel/exp/CALLAB/202130/p900203/raw"
+    ctrl_src = "FXE_DET_LPD1M-1/COMP/FEM_MDL_COMP"
+    runs = [9008, 9010, 9009]
+    sorted_runs = sort_dark_runs_by_gain(raw_folder, runs, ctrl_src)
+    assert sorted_runs == [9008, 9009, 9010]
+
+
+@pytest.mark.requires_gpfs
+def test_raise_sort_dark_runs_by_gain():
+    # TODO: update used raw data to test data when available.
+    raw_folder = "/gpfs/exfel/exp/CALLAB/202130/p900203/raw"
+    ctrl_src = "FXE_DET_LPD1M-1/COMP/FEM_MDL_COMP"
+    runs = [9008, 9009]
+    with pytest.raises(AssertionError) as err:
+        sort_dark_runs_by_gain(raw_folder, runs, ctrl_src)
+
+    runs = [9008, 9008, 9009]  # dynamicgain, dynamicgain, fixgain2
+    with pytest.raises(ValueError) as err:
+        sort_dark_runs_by_gain(raw_folder, runs, ctrl_src)
-- 
GitLab