import unittest

from extensions.chaeo.zstack import build_stack_mask
from model_server.accessors import InMemoryDataAccessor

class TestZStackDerivedDataProducts(unittest.TestCase):

    def setUp(self) -> None:
        # need test data incl obj map
        self.obmap = None
        self.stack = None

    def test_zmask_makes_correct_boxes(self):
        zmask, meta = build_stack_mask(
            'test_zmask_with boxes',
            self.stack,
            self.obmap,
            mask_type='boxes',
        )
        zmask_acc = InMemoryDataAccessor(zmask)
        self.assertTrue(zmask_acc.is_object_map())

        # assert dimensionality of zmask
        self.assertEqual(zmask.shape_dict['Z'] > 1)
        self.assertEqual(zmask.shape_dict['C'] == 1)

        # assert non-trivial meta info in boxes
        pass

    def test_zmask_makes_correct_contours(self):
        pass