Add dtype assertion to check_array_result for improved result validation

Author: PierreRaybaut

cdl/tests/features/images/profile_unit_test.py

@@ -7,6 +7,7 @@
 # pylint: disable=invalid-name  # Allows short reference names like x, y, ...
 # guitest: show
 
+import numpy as np
 import pytest
 from guidata.qthelpers import exec_dialog, qt_app_context
 
@@ -70,7 +71,7 @@ def test_line_profile() -> None:
     sig = cpi.compute_line_profile(ima, param)
     assert sig is not None
     assert len(sig.y) == width
-    exp = ima.data[row, :]
+    exp = np.array(ima.data[row, :], dtype=float)
     check_array_result("Horizontal line profile", sig.y, exp)
 
     # Test vertical line profile
@@ -79,7 +80,7 @@ def test_line_profile() -> None:
     sig = cpi.compute_line_profile(ima, param)
     assert sig is not None
     assert len(sig.y) == height
-    exp = ima.data[:, col]
+    exp = np.array(ima.data[:, col], dtype=float)
     check_array_result("Vertical line profile", sig.y, exp)
 
 
@@ -97,7 +98,7 @@ def test_segment_profile() -> None:
     sig = cpi.compute_segment_profile(ima, param)
     assert sig is not None
     assert len(sig.y) == min(row2, height - 1) - max(row1, 0) + 1
-    exp = ima.data[10:200, 20]
+    exp = np.array(ima.data[10:200, 20], dtype=float)
     check_array_result("Segment profile", sig.y, exp)
 
 
@@ -116,20 +117,20 @@ def test_average_profile() -> None:
     sig = cpi.compute_average_profile(ima, param)
     assert sig is not None
     assert len(sig.y) == col2 - col1 + 1
-    exp = ima.data[row1 : row2 + 1, col1 : col2 + 1].mean(axis=0)
+    exp = np.array(ima.data[row1 : row2 + 1, col1 : col2 + 1].mean(axis=0), dtype=float)
     check_array_result("Horizontal average profile", sig.y, exp)
 
     # Test vertical average profile
     param.direction = "vertical"
     sig = cpi.compute_average_profile(ima, param)
     assert sig is not None
     assert len(sig.y) == min(row2, height - 1) - max(row1, 0) + 1
-    exp = ima.data[row1 : row2 + 1, col1 : col2 + 1].mean(axis=1)
+    exp = np.array(ima.data[row1 : row2 + 1, col1 : col2 + 1].mean(axis=1), dtype=float)
     check_array_result("Vertical average profile", sig.y, exp)
 
 
 if __name__ == "__main__":
-    test_profile_unit()
+    # test_profile_unit()
     test_line_profile()
     test_segment_profile()
     test_average_profile()

cdl/utils/tests.py

@@ -283,6 +283,7 @@ def check_array_result(
     restxt = f"{title}: {__array_to_str(res)} (expected: {__array_to_str(exp)})"
     execenv.print(restxt)
     assert np.allclose(res, exp, rtol=rtol, atol=atol), restxt
+    assert res.dtype == exp.dtype, restxt
 
 
 def check_scalar_result(