add test_postprocessing_dtype

Author: FynnBe

tests/prediction_pipeline/test_combined_processing.py

@@ -0,0 +1,36 @@
+import numpy as np
+import xarray as xr
+
+from bioimageio.core.resource_io import nodes
+
+
+def test_postprocessing_dtype():
+    from bioimageio.core.prediction_pipeline._combined_processing import CombinedProcessing
+
+    shape = [3, 32, 32]
+    axes = ("c", "y", "x")
+    np_data = np.random.rand(*shape)
+    data = xr.DataArray(np_data, dims=axes)
+
+    inputs = []
+
+    threshold = 0.5
+    exp = xr.DataArray(np_data > threshold, dims=axes)
+
+    for dtype in ("float32", "float64", "uint8", "uint16"):
+        outputs = [
+            nodes.OutputTensor(
+                "out1",
+                data_type=dtype,
+                axes=axes,
+                shape=shape,
+                postprocessing=[nodes.Postprocessing("binarize", dict(threshold=threshold))],
+            )
+        ]
+        com_proc = CombinedProcessing(inputs, outputs)
+
+        res, _ = com_proc.apply_postprocessing(data, input_sample_statistics={})
+        assert len(res) == 1
+        res = res[0]
+        assert np.dtype(res.dtype) == np.dtype(dtype)
+        xr.testing.assert_allclose(res, exp.astype(dtype))