From 043195fec5bafcc88dba33e4be8f18b912662124 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Thu, 19 Mar 2026 16:49:34 +0000 Subject: [PATCH] fix: use format_tokens() in render_live_sessions (#133) Replace raw :, formatting with format_tokens() helper for consistent K/M suffix display across all views. Update test assertion and add M-suffix test for values >= 1,000,000. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --- src/copilot_usage/report.py | 2 +- tests/copilot_usage/test_report.py | 10 +++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/copilot_usage/report.py b/src/copilot_usage/report.py index 29c2eef..a38fbe0 100644 --- a/src/copilot_usage/report.py +++ b/src/copilot_usage/report.py @@ -156,7 +156,7 @@ def render_live_sessions(sessions: list[SessionSummary]) -> None: model = s.model or "—" running = _format_session_running_time(s) messages = str(s.user_messages) - tokens = f"{_estimated_output_tokens(s):,}" + tokens = format_tokens(_estimated_output_tokens(s)) cwd = s.cwd or "—" table.add_row( diff --git a/tests/copilot_usage/test_report.py b/tests/copilot_usage/test_report.py index 956a9c8..6615e18 100644 --- a/tests/copilot_usage/test_report.py +++ b/tests/copilot_usage/test_report.py @@ -190,7 +190,15 @@ def test_active_session_shows_output_tokens(self) -> None: output_tokens=15000, start_time=now - timedelta(minutes=5) ) output = _capture_output([session]) - assert "15,000" in output + assert "15.0K" in output + + def test_active_session_shows_output_tokens_m_suffix(self) -> None: + now = datetime.now(tz=UTC) + session = _make_session( + output_tokens=1_500_000, start_time=now - timedelta(minutes=5) + ) + output = _capture_output([session]) + assert "1.5M" in output def test_active_session_shows_cwd(self) -> None: now = datetime.now(tz=UTC)