From d67abf56083563eb9cdd5343b427cd4427303048 Mon Sep 17 00:00:00 2001
From: Sasa Junuzovic <44276455+microsasa@users.noreply.github.com>
Date: Fri, 10 Apr 2026 02:18:20 -0700
Subject: [PATCH] Add tests for IEEE 754 special floats in parse_token_int()

Add coverage for float('inf'), float('-inf'), float('nan'), and -0.0
in both parse_token_int() (TestParseTokenInt) and the Pydantic
boundary validator (TestSanitizeNonNumericTokens).

These values are already handled correctly by the implementation but
had no regression tests documenting the expected behavior.

Closes #895

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
---
 tests/copilot_usage/test_models.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/tests/copilot_usage/test_models.py b/tests/copilot_usage/test_models.py
index 3cd680f..26ae354 100644
--- a/tests/copilot_usage/test_models.py
+++ b/tests/copilot_usage/test_models.py
@@ -247,6 +247,17 @@ def test_whole_positive_float_coerced_to_int(self) -> None:
         d = AssistantMessageData.model_validate({"outputTokens": 100.0})
         assert d.outputTokens == 100
 
+    @pytest.mark.parametrize("raw", [float("inf"), float("-inf"), float("nan")])
+    def test_special_float_maps_to_zero(self, raw: float) -> None:
+        """IEEE 754 special floats on the Pydantic boundary must map to 0."""
+        d = AssistantMessageData.model_validate({"outputTokens": raw})
+        assert d.outputTokens == 0
+
+    def test_negative_zero_float_maps_to_zero(self) -> None:
+        """-0.0 on the Pydantic boundary must map to 0."""
+        d = AssistantMessageData.model_validate({"outputTokens": -0.0})
+        assert d.outputTokens == 0
+
 
 def test_session_shutdown_data() -> None:
     d = SessionShutdownData.model_validate(RAW_SHUTDOWN["data"])
@@ -407,6 +418,15 @@ def test_zero_or_negative_float_returns_none(self, value: float) -> None:
         """Zero and negative whole floats must be rejected."""
         assert parse_token_int(value) is None
 
+    @pytest.mark.parametrize("raw", [float("inf"), float("-inf"), float("nan")])
+    def test_special_float_returns_none(self, raw: float) -> None:
+        """IEEE 754 special floats must be rejected, not coerced to int."""
+        assert parse_token_int(raw) is None
+
+    def test_negative_zero_float_returns_none(self) -> None:
+        """-0.0 passes is_integer() but must not count as a positive token value."""
+        assert parse_token_int(-0.0) is None
+
     @pytest.mark.parametrize("value", [None, {}, [], object()])
     def test_other_types_return_none(self, value: object) -> None:
         """Unsupported types must be rejected."""