mergechannels 0.1.2__tar.gz → 0.2.0__tar.gz

{mergechannels-0.1.2 → mergechannels-0.2.0}/Cargo.lock

@@ -59,12 +59,13 @@ dependencies = [
 
 [[package]]
 name = "mergechannels"
-version = "0.1.2"
+version = "0.2.0"
 dependencies = [
  "lazy_static",
  "ndarray",
  "numpy",
  "pyo3",
+ "smallvec",
 ]
 
 [[package]]
@@ -239,6 +240,12 @@ version = "2.1.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "357703d41365b4b27c590e3ed91eabb1b663f07c4c084095e60cbed4362dff0d"
 
+[[package]]
+name = "smallvec"
+version = "1.15.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8917285742e9f3e1683f0a9c4e6b57960b7314d0b08d30d1ecd426713ee2eee9"
+
 [[package]]
 name = "syn"
 version = "2.0.98"

{mergechannels-0.1.2 → mergechannels-0.2.0}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "mergechannels"
-version = "0.1.2"
+version = "0.2.0"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
@@ -13,3 +13,4 @@ lazy_static = "1.5.0"
 ndarray = "0.16.1"
 numpy = "0.24.0"
 pyo3 = "0.24.1"
+smallvec = "1.15.0"

{mergechannels-0.1.2 → mergechannels-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mergechannels
-Version: 0.1.2
+Version: 0.2.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
@@ -12,7 +12,10 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: numpy>1.25.0
+Provides-Extra: nvim
 License-File: LICENSE
+Summary: Apply and merge colormaps
+Author-email: Zac Swider <zac.swider@gmail.com>
 License: MIT
 Requires-Python: >=3.9, <=3.13
 Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
@@ -60,8 +63,7 @@ plt.show()
 print(colorized.shape, colorized.dtype)
 >> (512, 512, 3) uint8
 ```
-![colorize a single image](https://raw.githubusercontent.com/zacswider/README_Images/main/camera_red-green.png
-)
+![colorize a single image](https://raw.githubusercontent.com/zacswider/README_Images/main/camera_red-green.png)
 
 
 ### apply a different colormap to each channel
@@ -71,38 +73,66 @@ import matplotlib.pyplot as plt
 import mergechannels as mc
 
 cells, nuclei = data.cells3d().max(axis=0)
-cells = to_uint8(cells)     # normalize your own arrays, mergechannels doesn't currently handle this
-nuclei = to_uint8(nuclei)   # normalize your own arrays, mergechannels doesn't currently handle this
+assert cells.dtype == 'uint16' and nuclei.dtype == 'uint16'
+fig, axes = plt.subplots(1, 2, figsize=(3, 6), dpi=300)
+for ax in axes.ravel(): ax.axis('off')
+(a, b) = axes.ravel()
+a.imshow(mc.merge([cells, nuclei],['Orange Hot', 'Cyan Hot']))
+b.imshow(mc.merge([cells, nuclei],['I Blue', 'I Forest'], blending='min'))
+fig.tight_layout()
+plt.show()
+```
+![max and min multicolor blending](https://raw.githubusercontent.com/zacswider/README_Images/main/overlay_normal_and_inverted.png)
 
-fig, axes = plt.subplots(2, 2, figsize=(6, 6), dpi=150)
+### apply a colormap to a whole stack
+```python
+from skimage import data
+from matplotlib import pyplot as plt
+import mergechannels as mc
+
+volume = data.cells3d()
+cells = volume[:, 0]
+nuclei = volume[:, 1]
+merged = mc.merge([cells, nuclei],['Orange Hot', 'Cyan Hot'])
+plt.imshow(merged[24]); plt.show()
+```
+![colorize a whole stack of images](https://raw.githubusercontent.com/zacswider/README_Images/main/merged_stacks.png)
+
+### adjust the saturation limits when applying colormaps
+``` python
+from skimage import data
+import matplotlib.pyplot as plt
+import mergechannels as mc
+
+cells, nuclei = data.cells3d().max(axis=0)
+channels = [cells, nuclei]
+colormaps = ['I Blue', 'I Forest']
+fig, axes = plt.subplots(1, 2, figsize=(3, 6), dpi=300)
 for ax in axes.ravel(): ax.axis('off')
-(a, b, c, d) = axes.ravel()
-a.imshow(cells, cmap='gray')
-b.imshow(nuclei, cmap='gray')
-c.imshow(
-    mc.merge(
-        [cells, nuclei],
-        ['Orange Hot', 'Cyan Hot'],  # maximum blending is the default
-    ),
-)
-d.imshow(
+(a, b) = axes.ravel()
+a.imshow(mc.merge(channels, colormaps, blending='min'))
+b.imshow(
     mc.merge(
-        [cells, nuclei],
-        ['I Blue', 'I Forest'],
-        blending='min',     # use minimum blending with inverted colormaps
+        channels,
+        colormaps,
+        blending='min',
+        saturation_limits=(
+            0.01,  # bottom 1% of pixels set to black point
+            0.97,  # top 3% of pixels set to white point
+        ),
     ),
 )
 fig.tight_layout()
 plt.show()
 ```
-![max and min multicolor blending](https://raw.githubusercontent.com/zacswider/README_Images/main/cells_multicolor.png)
+![adjust saturation limits](https://raw.githubusercontent.com/zacswider/README_Images/main/adjust_sat_lims.png)
 
 
 ## Roadmap
-mergechannels is currently incredibly simple. It can apply one or more colormaps to one or more 2D 8-bit images and that's it.
+mergechannels is currently incredibly simple. It can apply one or more colormaps to one or more 2D and 3D 8-bit or 16-bit images and that's it.
 - Add support for any numerical dtype
-- Add support for 3D images
 - Add option to return any colormap as a matplotlib colormap
+- Add option to pass external colormaps to mergechannels
 - Add support for directly passing matplotlib colormaps instead of colormap names
 - Parallelize colormap application on large images (if it's helpful)
 - Add option to overlay binary or instance masks onto colorized images

{mergechannels-0.1.2 → mergechannels-0.2.0}/README.md

@@ -41,8 +41,7 @@ plt.show()
 print(colorized.shape, colorized.dtype)
 >> (512, 512, 3) uint8
 ```
-![colorize a single image](https://raw.githubusercontent.com/zacswider/README_Images/main/camera_red-green.png
-)
+![colorize a single image](https://raw.githubusercontent.com/zacswider/README_Images/main/camera_red-green.png)
 
 
 ### apply a different colormap to each channel
@@ -52,38 +51,66 @@ import matplotlib.pyplot as plt
 import mergechannels as mc
 
 cells, nuclei = data.cells3d().max(axis=0)
-cells = to_uint8(cells)     # normalize your own arrays, mergechannels doesn't currently handle this
-nuclei = to_uint8(nuclei)   # normalize your own arrays, mergechannels doesn't currently handle this
+assert cells.dtype == 'uint16' and nuclei.dtype == 'uint16'
+fig, axes = plt.subplots(1, 2, figsize=(3, 6), dpi=300)
+for ax in axes.ravel(): ax.axis('off')
+(a, b) = axes.ravel()
+a.imshow(mc.merge([cells, nuclei],['Orange Hot', 'Cyan Hot']))
+b.imshow(mc.merge([cells, nuclei],['I Blue', 'I Forest'], blending='min'))
+fig.tight_layout()
+plt.show()
+```
+![max and min multicolor blending](https://raw.githubusercontent.com/zacswider/README_Images/main/overlay_normal_and_inverted.png)
 
-fig, axes = plt.subplots(2, 2, figsize=(6, 6), dpi=150)
+### apply a colormap to a whole stack
+```python
+from skimage import data
+from matplotlib import pyplot as plt
+import mergechannels as mc
+
+volume = data.cells3d()
+cells = volume[:, 0]
+nuclei = volume[:, 1]
+merged = mc.merge([cells, nuclei],['Orange Hot', 'Cyan Hot'])
+plt.imshow(merged[24]); plt.show()
+```
+![colorize a whole stack of images](https://raw.githubusercontent.com/zacswider/README_Images/main/merged_stacks.png)
+
+### adjust the saturation limits when applying colormaps
+``` python
+from skimage import data
+import matplotlib.pyplot as plt
+import mergechannels as mc
+
+cells, nuclei = data.cells3d().max(axis=0)
+channels = [cells, nuclei]
+colormaps = ['I Blue', 'I Forest']
+fig, axes = plt.subplots(1, 2, figsize=(3, 6), dpi=300)
 for ax in axes.ravel(): ax.axis('off')
-(a, b, c, d) = axes.ravel()
-a.imshow(cells, cmap='gray')
-b.imshow(nuclei, cmap='gray')
-c.imshow(
-    mc.merge(
-        [cells, nuclei],
-        ['Orange Hot', 'Cyan Hot'],  # maximum blending is the default
-    ),
-)
-d.imshow(
+(a, b) = axes.ravel()
+a.imshow(mc.merge(channels, colormaps, blending='min'))
+b.imshow(
     mc.merge(
-        [cells, nuclei],
-        ['I Blue', 'I Forest'],
-        blending='min',     # use minimum blending with inverted colormaps
+        channels,
+        colormaps,
+        blending='min',
+        saturation_limits=(
+            0.01,  # bottom 1% of pixels set to black point
+            0.97,  # top 3% of pixels set to white point
+        ),
     ),
 )
 fig.tight_layout()
 plt.show()
 ```
-![max and min multicolor blending](https://raw.githubusercontent.com/zacswider/README_Images/main/cells_multicolor.png)
+![adjust saturation limits](https://raw.githubusercontent.com/zacswider/README_Images/main/adjust_sat_lims.png)
 
 
 ## Roadmap
-mergechannels is currently incredibly simple. It can apply one or more colormaps to one or more 2D 8-bit images and that's it.
+mergechannels is currently incredibly simple. It can apply one or more colormaps to one or more 2D and 3D 8-bit or 16-bit images and that's it.
 - Add support for any numerical dtype
-- Add support for 3D images
 - Add option to return any colormap as a matplotlib colormap
+- Add option to pass external colormaps to mergechannels
 - Add support for directly passing matplotlib colormaps instead of colormap names
 - Parallelize colormap application on large images (if it's helpful)
 - Add option to overlay binary or instance masks onto colorized images

{mergechannels-0.1.2 → mergechannels-0.2.0}/pyproject.toml

@@ -4,6 +4,9 @@ build-backend = "maturin"
 
 [project]
 name = "mergechannels"
+description = "Apply and merge colormaps"
+readme = "README.md"
+authors = [{ name = "Zac Swider", email = "zac.swider@gmail.com" }]
 license = { text = "MIT" }
 requires-python = ">=3.9, <=3.13"
 classifiers = [
@@ -20,6 +23,9 @@ classifiers = [
 ]
 dynamic = ["version"]
 dependencies = ["numpy>1.25.0"]
+
+[project.optional-dependencies]
+nvim = []
 [tool.maturin]
 features = ["pyo3/extension-module"]
 python-source = "python"
@@ -28,6 +34,7 @@ python-source = "python"
 dev = [
     "matplotlib>=3.7.5",
     "pre-commit>=3.5.0",
+    "pynvim>=0.5.2",
     "pytest-benchmark>=4.0.0",
     "pytest>=8.3.5",
 ]

{mergechannels-0.1.2 → mergechannels-0.2.0}/pytests/test_colorize.py

@@ -7,7 +7,7 @@ def test_apply_color_map():
     Test that the color map is applied correctly
     '''
     x = np.ones((1, 1), dtype=np.uint8)
-    rgb = mc.apply_color_map(x, 'betterBlue')
+    rgb = mc.dispatch_single_channel(x, 'betterBlue', (0, 255))
     assert rgb.shape == (1,1,3)
     assert rgb.dtype == np.uint8
     assert np.allclose(
@@ -17,7 +17,7 @@ def test_apply_color_map():
         )
     )
     x = np.ones((1, 1), dtype=np.uint8) * 255
-    rgb = mc.apply_color_map(x, 'betterBlue')
+    rgb = mc.dispatch_single_channel(x, 'betterBlue', (0, 255))
     assert np.allclose(
         rgb,
         np.array(
@@ -31,7 +31,12 @@ def test_apply_colors_and_merge_low_sum():
     '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
-    rgb_sum = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'sum')
+    rgb_sum = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'sum',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 1, 2,]
     # orange = [1, 1, 0]
     assert np.allclose(
@@ -47,7 +52,12 @@ def test_apply_colors_and_merge_high_sum():
     '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
-    rgb_sum = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'sum')
+    rgb_sum = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'sum',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 188, 254]
     # orange = [255, 149, 0]
     assert np.allclose(
@@ -63,7 +73,12 @@ def test_apply_colors_and_merge_low_max():
     '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
-    rgb_max = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'max')
+    rgb_max = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'max',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 1, 2,]
     # orange = [1, 1, 0]
     assert np.allclose(
@@ -79,7 +94,12 @@ def test_apply_colors_and_merge_high_max():
     '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
-    rgb_max = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'max')
+    rgb_max = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'max',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 188, 254]
     # orange = [255, 149, 0]
     assert np.allclose(
@@ -95,7 +115,12 @@ def test_apply_colors_and_merge_low_min():
     '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
-    rgb_min = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'min')
+    rgb_min = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'min',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 1, 2,]
     # orange = [1, 1, 0]
     assert np.allclose(
@@ -111,7 +136,12 @@ def test_apply_colors_and_merge_high_min():
     '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
-    rgb_min = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'min')
+    rgb_min = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'min',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 188, 254]
     # orange = [255, 149, 0]
     assert np.allclose(
@@ -127,7 +157,12 @@ def test_apply_colors_and_merge_low_mean():
     '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
-    rgb_mean = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'mean')
+    rgb_mean = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'mean',
+        [(0, 255), (0, 255)],
+    )
     # blue = [0, 1, 2,]
     # orange = [1, 1, 0]
     assert np.allclose(
@@ -145,7 +180,12 @@ def test_apply_colors_and_merge_high_mean():
     '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
-    rgb_mean = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'mean')
+    rgb_mean = mc.dispatch_multi_channel(
+        [x, y],
+        ['betterBlue', 'betterOrange'],
+        'mean',
+        [(0, 255), (0, 255)]
+    )
     # blue = [0, 188, 254]
     # orange = [255, 149, 0]
     assert np.allclose(

{mergechannels-0.1.2 → mergechannels-0.2.0}/python/mergechannels/__init__.py

@@ -1,13 +1,13 @@
 # Import Rust functions
 from .mergechannels import (  # type: ignore
-	apply_color_map,
-	apply_colors_and_merge_nc,
+	dispatch_single_channel,
+	dispatch_multi_channel,
 )
-
 from ._internal import merge
 
+
 __all__ = [
-	'apply_color_map',
-	'apply_colors_and_merge_nc',
+	'dispatch_single_channel',
+	'dispatch_multi_channel',
 	'merge',
 ]

mergechannels-0.2.0/python/mergechannels/__init__.pyi

@@ -0,0 +1,30 @@
+
+from typing import Literal, Sequence
+import numpy as np
+
+from python.mergechannels._blending import BLENDING_OPTIONS
+from python.mergechannels._luts import COLORMAPS
+
+
+def merge(
+    arrs: np.ndarray | Sequence[np.ndarray],
+    colors: Sequence[COLORMAPS] = (),
+    blending: Literal[BLENDING_OPTIONS] = 'max',
+    saturation_limits: tuple[float, float] = (0.2, 99.8),
+) -> np.ndarray:
+    ...
+
+def dispatch_single_channel(
+    array_reference: np.ndarray,
+    cmap_name: str,
+    limits: tuple[float, float],
+) -> np.ndarray:
+    ...
+
+def dispatch_multi_channel(
+    array_references: Sequence[np.ndarray],
+    cmap_names: Sequence[str],
+    blending: Literal[BLENDING_OPTIONS],
+    limits: Sequence[tuple[float, float]],
+) -> np.ndarray:
+    ...

{mergechannels-0.1.2 → mergechannels-0.2.0}/python/mergechannels/_internal.py

@@ -3,8 +3,8 @@ from typing import Sequence
 import numpy as np
 
 from mergechannels import (
-	apply_color_map,
-	apply_colors_and_merge_nc,
+	dispatch_single_channel,
+	dispatch_multi_channel,
 )
 from ._luts import COLORMAPS
 from ._blending import BLENDING_OPTIONS
@@ -13,6 +13,7 @@ def merge(
 	arrs: Sequence[np.ndarray],
 	colors: Sequence[COLORMAPS],
 	blending: BLENDING_OPTIONS = 'max',
+    saturation_limits: tuple[float, float] = (0.011, 0.999),
 ) -> np.ndarray:
 	'''
 	apply cmaps to arrays and blend the colors
@@ -30,9 +31,9 @@ def merge(
 		raise ValueError(
 			f'Expected every array to have the same shape, got {arr_shapes}'
 		)
-	if not len(arr_shapes[0]) == 2:
+	if len(arr_shapes[0]) not in (2, 3):
 		raise ValueError(
-			f'Expected every array to be 2D, got {arr_shapes[0]}'
+			f'Expected every array to be 2D or 3D, got {arr_shapes[0]}'
 		)
 	arr_dtypes = [arr.dtype for arr in arrs]
 	if not len(set(arr_dtypes)) == 1:
@@ -40,8 +41,25 @@ def merge(
 			f'Expected every array to have the same dtype, got {arr_dtypes}'
 		)
 	# endregion
-
 	if n_arrs == 1:
-		return apply_color_map(arr=arrs[0], cmap_name=colors[0])
+		if arrs[0].dtype == 'uint8':
+			limits = (0, 255)
+		else:
+			low, high = np.percentile(arrs[0], np.array(saturation_limits) * 100)
+			limits = (low, high)
+		return dispatch_single_channel(
+			array_reference=arrs[0],
+			cmap_name=colors[0],
+			limits=limits,
+		)
 	else:
-		return apply_colors_and_merge_nc(arrs, colors, blending)
+		if all(arr.dtype == 'uint8' for arr in arrs):
+			limits = (0, 255)
+		else:
+			limits = tuple(np.percentile(arr, np.array(saturation_limits) * 100) for arr in arrs)
+		return dispatch_multi_channel(
+			array_references=arrs,
+			cmap_names=colors,
+			blending=blending,
+			limits=limits,  # type: ignore
+		)

{mergechannels-0.1.2 → mergechannels-0.2.0}/src/blend.rs

@@ -1,4 +1,33 @@
-pub fn max_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
+use smallvec::SmallVec;
+use std::error::Error;
+use std::fmt;
+
+pub const MAX_N_CH: usize = 5;
+
+pub type BlendFn = fn(&SmallVec<[[u8; 3]; 5]>) -> [u8; 3];
+
+#[derive(Debug)]
+pub enum MergeError {
+    InvalidBlendingMode(String),
+}
+
+impl fmt::Display for MergeError {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            MergeError::InvalidBlendingMode(mode) => {
+                write!(
+                    f,
+                    "Invalid blending mode: `{}`. Valid modes are 'max', 'sum', 'min', and 'mean'.",
+                    mode
+                )
+            }
+        }
+    }
+}
+
+impl Error for MergeError {}
+
+pub fn max_blending(px_vals: &SmallVec<[[u8; 3]; 5]>) -> [u8; 3] {
     let mut r: u8 = 0;
     let mut g: u8 = 0;
     let mut b: u8 = 0;
@@ -16,7 +45,7 @@ pub fn max_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
     [r, g, b]
 }
 
-pub fn sum_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
+pub fn sum_blending(px_vals: &SmallVec<[[u8; 3]; 5]>) -> [u8; 3] {
     let mut r: u8 = 0;
     let mut g: u8 = 0;
     let mut b: u8 = 0;
@@ -28,7 +57,7 @@ pub fn sum_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
     [r, g, b]
 }
 
-pub fn min_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
+pub fn min_blending(px_vals: &SmallVec<[[u8; 3]; 5]>) -> [u8; 3] {
     let mut r: u8 = 255;
     let mut g: u8 = 255;
     let mut b: u8 = 255;
@@ -46,7 +75,7 @@ pub fn min_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
     [r, g, b]
 }
 
-pub fn mean_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
+pub fn mean_blending(px_vals: &SmallVec<[[u8; 3]; 5]>) -> [u8; 3] {
     let mut r: u16 = 0;
     let mut g: u16 = 0;
     let mut b: u16 = 0;
@@ -65,31 +94,37 @@ pub fn mean_blending(px_vals: &Vec<[u8; 3]>) -> [u8; 3] {
 #[cfg(test)]
 mod tests {
     use super::*;
+    fn create_test_data() -> SmallVec<[[u8; 3]; MAX_N_CH]> {
+        let mut px_vals: SmallVec<[[u8; 3]; MAX_N_CH]> = SmallVec::new();
+        px_vals.push([100, 100, 100]);
+        px_vals.push([200, 200, 200]);
+        px_vals
+    }
 
     #[test]
     fn test_max_blending() {
-        let px_vals = vec![[100, 100, 100], [200, 200, 200]];
+        let px_vals = create_test_data();
         let result = max_blending(&px_vals);
         assert_eq!(result, [200, 200, 200]);
     }
 
     #[test]
     fn test_sum_blending() {
-        let px_vals = vec![[100, 100, 100], [200, 200, 200]];
+        let px_vals = create_test_data();
         let result = sum_blending(&px_vals);
         assert_eq!(result, [255, 255, 255]);
     }
 
     #[test]
     fn test_min_blending() {
-        let px_vals = vec![[100, 100, 100], [200, 200, 200]];
+        let px_vals = create_test_data();
         let result = min_blending(&px_vals);
         assert_eq!(result, [100, 100, 100]);
     }
 
     #[test]
     fn test_mean_blending() {
-        let px_vals = vec![[100, 100, 100], [200, 200, 200]];
+        let px_vals = create_test_data();
         let result = mean_blending(&px_vals);
         assert_eq!(result, [150, 150, 150]);
     }