Enhance inverse computation tests to handle infinite values and assert NaN equality

Author: PierreRaybaut

cdl/tests/features/images/operation_unit_test.py

@@ -231,6 +231,7 @@ def test_image_inverse() -> None:
         with warnings.catch_warnings():
             warnings.simplefilter("ignore", category=RuntimeWarning)
             exp = np.reciprocal(ima1.data, dtype=float)
+            exp[np.isinf(exp)] = np.nan
         ima2 = cpi.compute_inverse(ima1)
         check_array_result("Image inverse", ima2.data, exp)
 

cdl/tests/features/signals/operation_unit_test.py

@@ -13,6 +13,8 @@
 
 from __future__ import annotations
 
+import warnings
+
 import numpy as np
 import pytest
 
@@ -120,7 +122,11 @@ def test_signal_inverse() -> None:
     """Signal inversion validation test."""
     s1 = __create_two_signals()[0]
     inv_signal = cps.compute_inverse(s1)
-    check_array_result("Signal inverse", inv_signal.y, 1.0 / s1.y)
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore", category=RuntimeWarning)
+        exp = 1.0 / s1.y
+        exp[np.isinf(exp)] = np.nan
+    check_array_result("Signal inverse", inv_signal.y, exp)
 
 
 @pytest.mark.validation

cdl/utils/tests.py

@@ -282,7 +282,7 @@ def check_array_result(
     """Assert that two arrays are almost equal."""
     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 np.allclose(res, exp, rtol=rtol, atol=atol, equal_nan=True), restxt
     assert res.dtype == exp.dtype, restxt