From 71ce1f57e7e6dd4a44f859f5365e23308a7fdb56 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 23 Sep 2024 11:33:46 +0200
Subject: [PATCH] Repaired a few things that were causing ilastik test failures

---
 tests/test_ilastik/test_ilastik.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/test_ilastik/test_ilastik.py b/tests/test_ilastik/test_ilastik.py
index 35a41813..fd6893a5 100644
--- a/tests/test_ilastik/test_ilastik.py
+++ b/tests/test_ilastik/test_ilastik.py
@@ -363,7 +363,7 @@ class TestIlastikOnMultichannelInputs(conf.TestServerBaseClass):
         acc_input = generate_file_accessor(self.pa_input_image)
         acc_obmap = generate_file_accessor(res.object_map_filepath)
         self.assertEqual(acc_obmap.hw, acc_input.hw)
-        self.assertEqual(len(acc_obmap._unique()[1]), 3)
+        self.assertEqual(len(acc_obmap.unique()[1]), 3)
 
 
     def test_api(self):
@@ -401,9 +401,9 @@ class TestIlastikObjectClassification(unittest.TestCase):
         stack_ch_pa = stack.get_mono(conf.meta['roiset']['patches_channel'])
         seg_mask = generate_file_accessor(data['multichannel_zstack_mask2d']['path'])
 
-        self.roiset = RoiSet(
+        self.roiset = RoiSet.from_binary_mask(
             stack_ch_pa,
-            get_label_ids(seg_mask),
+            seg_mask,
             params=RoiSetMetaParams(
                 mask_type='boxes',
                 filters={'area': {'min': 1e3, 'max': 1e4}},
-- 
GitLab