From 6364418b84c0ae3b95d9ba2b85ba17089fb298ba Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Tue, 10 Oct 2023 10:10:05 +0200
Subject: [PATCH] Now test against multichannel TIF that includes chlorophyl

---
 extensions/chaeo/conf/testing.py      | 4 ++--
 extensions/chaeo/tests/test_zstack.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/extensions/chaeo/conf/testing.py b/extensions/chaeo/conf/testing.py
index 81a0ba82..4ca90680 100644
--- a/extensions/chaeo/conf/testing.py
+++ b/extensions/chaeo/conf/testing.py
@@ -3,10 +3,10 @@ from pathlib import Path
 root = Path.home() / 'model-server'
 
 multichannel_zstack = {
-    'path': root / 'testing' / 'zmask-test-stack.tif',
+    'path': root / 'testing' / 'zmask-test-stack-chlorophyl.tif',
     'w': 512,
     'h': 512,
-    'c': 2,
+    'c': 5,
     'z': 7,
 }
 
diff --git a/extensions/chaeo/tests/test_zstack.py b/extensions/chaeo/tests/test_zstack.py
index c2d0b113..92d3ef25 100644
--- a/extensions/chaeo/tests/test_zstack.py
+++ b/extensions/chaeo/tests/test_zstack.py
@@ -86,7 +86,7 @@ class TestZStackDerivedDataProducts(unittest.TestCase):
         )
         files = export_patches_from_zstack(
             output_path / '3d_patches',
-            self.stack.get_one_channel_data(channel=1),
+            self.stack.get_one_channel_data(4),
             meta,
             make_3d=True)
         self.assertGreaterEqual(len(files), 1)
@@ -94,7 +94,7 @@ class TestZStackDerivedDataProducts(unittest.TestCase):
     def test_flatten_image(self):
         zmask, meta, df, interm = build_zmask_from_object_mask(
             self.obmap.get_one_channel_data(0),
-            self.stack.get_one_channel_data(1),
+            self.stack.get_one_channel_data(4),
             mask_type='boxes',
         )
 
-- 
GitLab