diff --git a/model_server/extensions/ilastik/tests/test_ilastik.py b/model_server/extensions/ilastik/tests/test_ilastik.py index 18e95a4df44279b452bc11226700375b63c15381..680b67c948d9624c48606944f345a88c01ff6e64 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(