Confirms that ilastik pixel object already works with multichannel input

model_server/extensions/ilastik/tests/test_ilastik.py

@@ -344,6 +344,15 @@ class TestIlastikOnMultichannelInputs(TestServerBaseClass):
         mod = ilm.IlastikPixelClassifierModel(params={'project_file': self.pa_px_classifier})
         pxmap = mod.infer(self.input_image)[0]
         self.assertEqual(pxmap.hw, self.input_image.hw)
+        self.assertEqual(pxmap.nz, self.input_image.nz)
+        return pxmap
+
+    def test_classify_objects(self):
+        pxmap = self.test_classify_pixels()
+        mod = ilm.IlastikObjectClassifierFromPixelPredictionsModel(params={'project_file': self.pa_ob_classifier})
+        obmap = mod.infer(self.input_image, pxmap)[0]
+        self.assertEqual(obmap.hw, self.input_image.hw)
+        self.assertEqual(obmap.nz, self.input_image.nz)