fix(test_truncate): add test to make sure negative values are correct…

…ly truncated with relative precision
radiocosmology · Dec 18, 2024 · d86f62c · d86f62c
1 parent f8d4956
commit d86f62c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_truncate.py b/tests/test_truncate.py
@@ -126,3 +126,19 @@ def test_truncate_relative():
         )
         == np.asarray([32, 32], dtype=np.float64)
     ).all()
+
+    # Check the case where values are negative
+    assert (
+        truncate.bit_truncate_relative(
+            np.asarray([-32.121, 32.5], dtype=np.float32),
+            0.1,
+        )
+        == np.asarray([-32, 32], dtype=np.float32)
+    ).all()
+    assert (
+        truncate.bit_truncate_relative(
+            np.asarray([-32.121, 32.5], dtype=np.float64),
+            0.1,
+        )
+        == np.asarray([-32, 32], dtype=np.float64)
+    ).all()