Fix for replacing NaN values

Author: juntyr

src/numcodecs_replace/__init__.py

@@ -153,7 +153,10 @@ def encode(self, buf: Buffer) -> Buffer:
         }
 
         for k, v in replacements.items():
-            a[a == k] = v
+            if isinstance(k, int) or not np.isnan(k):
+                a[a == k] = v
+            else:
+                a[np.isnan(a)] = v
 
         return a  # type: ignore
 

tests/test_replace.py

@@ -81,6 +81,7 @@ def test_roundtrip():
     with pytest.warns(RuntimeWarning, match="empty slice"):
         check_roundtrip(np.zeros((0,)))
     check_roundtrip(np.arange(1000).reshape(10, 10, 10))
+    check_roundtrip(np.array([4.2, -2.4, np.nan, -np.nan, 0.0, -0.0]))
     check_roundtrip(np.array([np.inf, -np.inf, np.nan, -np.nan, 0.0, -0.0]))
     check_roundtrip(
         np.array(