Properly deal with NaNs when comparing sparse and dense matrices

paulsengroup · Oct 15, 2024 · fef0d45 · fef0d45
1 parent ee1f1c6
commit fef0d45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/scripts/fuzzer.py b/test/scripts/fuzzer.py
@@ -263,7 +263,7 @@ def compare_numpy(
         )
         return False
 
-    num_differences = (~np.isclose(expected, found, rtol=rtol)).sum()
+    num_differences = (~np.isclose(expected, found, rtol=rtol, equal_nan=True)).sum()
     if num_differences != 0:
         logging.warning(
             "[%d] %s, %s (%d nnz): FAIL! Found %d differences!",
@@ -333,7 +333,7 @@ def compare_csr(
         )
         return False
 
-    num_differences = (~np.isclose(expected.data, found.data, rtol=rtol)).sum()
+    num_differences = (~np.isclose(expected.data, found.data, rtol=rtol, equal_nan=True)).sum()
     if num_differences != 0:
         logging.warning(
             "[%d] %s, %s (%d nnz): FAIL! Found %d differences!",