From 6bc4b7e0c89d023289a954544484e226d9f947b6 Mon Sep 17 00:00:00 2001
From: Kris Thielemans <k.thielemans@ucl.ac.uk>
Date: Sun, 29 Sep 2024 16:18:50 +0100
Subject: [PATCH] [CI] minimal check on get_data()

---
 .github/workflows/run.yml                   | 5 +++++
 SIRF_data_preparation/plot_BSREM_metrics.py | 6 +++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/run.yml b/.github/workflows/run.yml
index 58bf984..97fc412 100644
--- a/.github/workflows/run.yml
+++ b/.github/workflows/run.yml
@@ -36,11 +36,16 @@ jobs:
         curl -fsSL https://raw.githubusercontent.com/SyneRBI/PETRIC/main/petric.py > petric.py
         test -f main.py || ln -s main_ISTA.py main.py
         python <<EOF
+        import petric
         from main import Submission, submission_callbacks
         from cil.optimisation.algorithms import Algorithm
         assert issubclass(Submission, Algorithm)
         assert isinstance(submission_callbacks, list)
         assert all(callable(callback) for callback in submission_callbacks)
+        data = petric.get_data()
+        assert data.OSEM_image.max() > 0
+        assert data.reference_image.max() > 0
+        assert data.FOV_mask.max() > 0
         EOF
   full:
     if: startsWith(github.ref, 'refs/tags')
diff --git a/SIRF_data_preparation/plot_BSREM_metrics.py b/SIRF_data_preparation/plot_BSREM_metrics.py
index 39bb828..74c2273 100644
--- a/SIRF_data_preparation/plot_BSREM_metrics.py
+++ b/SIRF_data_preparation/plot_BSREM_metrics.py
@@ -23,7 +23,9 @@
 # scanID = 'Siemens_Vision600_thorax'
 # scanID = 'NeuroLF_Hoffman_Dataset'
 # scanID = 'Siemens_mMR_NEMA_IQ'
-scanID = 'Mediso_NEMA_IQ'
+# scanID = 'Mediso_NEMA_IQ'
+# scanID = 'GE_DMI3_Torso'
+scanID = 'Siemens_mMR_NEMA_IQ_lowcounts'
 
 srcdir = SRCDIR / scanID
 outdir = OUTDIR / scanID
@@ -126,3 +128,5 @@
 plt.figure()
 data_QC.plot_image(image - reference_image, **slices, vmin=-cmax / 100, vmax=cmax / 100)
 plt.savefig(outdir / f'{scanID}_ref_diff_image_at_0.01_0.005.png')
+
+# %%