Test covers multiple raw and multiple derived channels

tests/test_roiset.py

@@ -217,9 +217,7 @@ class TestRoiSetMonoProducts(BaseTestRoiSetMonoProducts, unittest.TestCase):
             self.stack,
             self.seg_mask,
             params=RoiSetMetaParams(
-                # mask_type=mask_type,
                 filters={'area': {'min': 1e3, 'max': 1e4}},
-                # expand_box_by=(128, 2)
             )
         )
         roiset.classify_by(
@@ -231,12 +229,14 @@ class TestRoiSetMonoProducts(BaseTestRoiSetMonoProducts, unittest.TestCase):
                 lambda acc: PatchStack((0.5 * acc.get_channels([1]).data).astype('uint8'))
             ]
         )
-        self.assertTrue(all(roiset.get_df()['classify_by_multiple_input_model'].unique() == [3]))
-        self.assertTrue(all(np.unique(roiset.object_class_maps['multiple_input_model'].data) == [0, 3]))
+        self.assertTrue(all(roiset.get_df()['classify_by_multiple_input_model'].unique() == [4]))
+        self.assertTrue(all(np.unique(roiset.object_class_maps['multiple_input_model'].data) == [0, 4]))
 
         self.assertEqual(len(roiset.accs_derived), 2)
         for di in roiset.accs_derived:
-            self.assertEqual(roiset.get_patches_acc().shape, di.shape)
+            self.assertEqual(roiset.get_patches_acc().hw, di.hw)
+            self.assertEqual(roiset.get_patches_acc().nz, di.nz)
+            self.assertEqual(roiset.get_patches_acc().count, di.count)
 
         dpas = roiset.run_exports(output_path / 'derived_channels', 0, 'der', RoiSetExportParams(derived_channels=True))
         for fp in dpas['derived_channels']: