Testing save colored mask

Author: alex-l-kong

tests/utils/plot_utils_test.py

@@ -800,8 +800,14 @@ def test_save_colored_masks(
         # check that colored mask is mapped correctly
         cluster_mask = io.imread(os.path.join(create_masks, fov + f'_{cluster_type}_mask.tiff'))
         rgb_mask = (colored_mask[:, :, 0]/255).round(1)
+        print(list(metacluster_colors.keys()))
 
         for id_num in metacluster_colors.keys():
+            cluster_idx = np.where(cluster_mask == id_num)
+            colored_idx = np.where(rgb_mask == np.round(metacluster_colors[id_num][0], 1))
+            print(id_num)
+            print(cluster_idx)
+            print(colored_idx)
             if id_num != 6:
                 cluster_idx = np.where(cluster_mask == id_num)
                 colored_idx = np.where(rgb_mask == np.round(metacluster_colors[id_num][0], 1))