diff --git a/extensions/chaeo/test_zstack.py b/extensions/chaeo/test_zstack.py index 78baa64c50f6a5ef167c261f63064199e6840fb9..3a517907dcc1e0e344df3eeffe57f035f3da0c1b 100644 --- a/extensions/chaeo/test_zstack.py +++ b/extensions/chaeo/test_zstack.py @@ -2,13 +2,19 @@ import unittest from extensions.chaeo.conf.testing import multichannel_zstack, pixel_classifier from extensions.chaeo.zstack import build_stack_mask -from model_server.accessors import InMemoryDataAccessor +from model_server.accessors import generate_file_accessor, InMemoryDataAccessor +from extensions.ilastik.models import IlastikObjectClassifierModel, IlastikPixelClassifierModel class TestZStackDerivedDataProducts(unittest.TestCase): def setUp(self) -> None: # need test data incl obj map - self.zstack = + self.zstack = generate_file_accessor(multichannel_zstack['path']) + + pxmodel = IlastikPixelClassifierModel( + project_file=pixel_classifier['path'] + ) + self.pxmap = None self.obmap = None self.stack = None