Fix test

Signed-off-by: Joaquin Anton <[email protected]>

Author: jantonguirao

dali/test/python/operator_1/test_debayer.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright (c) 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,6 +22,7 @@
 from test_utils import get_dali_extra_path
 from nose_utils import assert_raises
 from nose2.tools import params
+from test_utils import dali_type
 from debayer_test_utils import (
     bayer_patterns,
     blue_position,
@@ -42,12 +43,12 @@ def read_imgs(num_imgs, dtype, seed):
     @pipeline_def
     def pipeline():
         input, _ = fn.readers.file(file_root=images_dir, random_shuffle=True, seed=seed)
-        return fn.decoders.image(input, device="cpu", output_type=types.RGB)
+        return fn.decoders.image(input, device="cpu", output_type=types.RGB, dtype=dali_type(dtype))
 
     pipe = pipeline(batch_size=num_imgs, device_id=0, num_threads=4)
     pipe.build()
     (batch,) = pipe.run()
-    return [np.array(img, dtype=dtype) for img in batch]
+    return [np.array(img) for img in batch]
 
 
 def read_video(num_sequences, num_frames, height, width, seed=42):
@@ -81,10 +82,6 @@ def prepare_test_imgs(num_samples, dtype):
     assert dtype in (np.uint8, np.uint16)
     rng = np.random.default_rng(seed=101)
     imgs = read_imgs(num_samples, dtype, seed=42 if dtype == np.uint8 else 13)
-    if dtype == np.uint16:
-        imgs = [
-            np.uint16(img) * 256 + np.uint16(rng.uniform(0, 256, size=img.shape)) for img in imgs
-        ]
     bayered_imgs = {
         pattern: [rgb2bayer(img, pattern) for img in imgs] for pattern in bayer_patterns
     }
@@ -154,7 +151,7 @@ def debayer_pipeline():
             assert len(debayered_imgs) == len(idxs)
             for img_debayered, idx in zip(debayered_imgs, idxs):
                 baseline = npp_baseline[pattern][idx]
-                assert np.all(img_debayered == baseline)
+                np.testing.assert_allclose(img_debayered, baseline)
 
     @params(*itertools.product([1, 11, 184], [np.uint8, np.uint16]))
     def test_debayer_per_sample_pattern(self, batch_size, dtype):
@@ -200,7 +197,7 @@ def debayer_pipeline():
             assert len(debayered_imgs) == len(patterns) == len(idxs)
             for img_debayered, pattern, idx in zip(debayered_imgs, patterns, idxs):
                 baseline = npp_baseline[pattern][idx]
-                assert np.all(img_debayered == baseline)
+                np.testing.assert_allclose(img_debayered, baseline)
 
 
 class DebayerVideoTest(unittest.TestCase):
@@ -260,7 +257,7 @@ def debayer_pipeline():
             assert len(debayered_videos) == len(idxs)
             for vid_debayered, idx in zip(debayered_videos, idxs):
                 baseline = self.npp_baseline[idx]
-                assert np.all(vid_debayered == baseline)
+                np.testing.assert_allclose(vid_debayered, baseline)
 
 
 def source_full_array(shape, dtype):