From 9db2ea507c99a0e077d5d5d62e19ab2abb5dcd9f Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Fri, 26 Apr 2024 15:50:49 +0200
Subject: [PATCH] ilastik workflow now interprets channel=None to use all input
 channels

---
 .../extensions/ilastik/tests/test_ilastik.py  | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/model_server/extensions/ilastik/tests/test_ilastik.py b/model_server/extensions/ilastik/tests/test_ilastik.py
index 18e95a4d..680b67c9 100644
--- a/model_server/extensions/ilastik/tests/test_ilastik.py
+++ b/model_server/extensions/ilastik/tests/test_ilastik.py
@@ -377,6 +377,26 @@ class TestIlastikOnMultichannelInputs(TestServerBaseClass):
         self.assertEqual(acc_obmap.hw, acc_input.hw)
         self.assertEqual(len(acc_obmap._unique()[1]), 3)
 
+    def test_roiset(self): # TODO: check on how RoiSet is interpreting 3D segmentation instances
+        img = generate_file_accessor(self.pa_input_image)
+        seg = generate_file_accessor(self.pa_input_image)
+
+        roiset = RoiSet(
+            img,
+            _get_label_ids(seg),
+            params=RoiSetMetaParams(
+                mask_type='boxes',
+                filters={'area': {'min': 1e3, 'max': 1e4}},
+                expand_box_by=(64, 2)
+            )
+        )
+        roiset.classify_by(
+            'multichannel_object_class',
+            range(0, img.chroma),
+            ilm.IlastikObjectClassifierFromSegmentationModel(params={'project_file': self.pa_ob_seg_classifier}),
+        )
+        self.assertTrue(False)
+
 
     def test_api(self):
         resp_load = self._put(
-- 
GitLab