fix(stats): remove unused imports from test_stats_collector

Author: elkaix

tests/ui_and_conv/test_stats_collector.py

@@ -1,18 +1,13 @@
 from __future__ import annotations
 
 import json
-import tempfile
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from pathlib import Path
 
-import pytest
-
 from pythinker_code.ui.shell.stats_collector import (
     StepRecord,
-    UsagePeriod,
     collect_session_files,
     compute_period_stats,
-    compute_insights,
     get_sessions_root,
     parse_wire_file,
 )
@@ -27,10 +22,15 @@ def _make_wire(tmp_path: Path, records: list[dict]) -> Path:
     return p
 
 
-def _status_update(ts: float, input_other: int, output: int,
-                   cache_read: int = 0, cache_write: int = 0,
-                   model_name: str | None = None,
-                   provider_key: str | None = None) -> dict:
+def _status_update(
+    ts: float,
+    input_other: int,
+    output: int,
+    cache_read: int = 0,
+    cache_write: int = 0,
+    model_name: str | None = None,
+    provider_key: str | None = None,
+) -> dict:
     return {
         "timestamp": ts,
         "message": {
@@ -50,11 +50,14 @@ def _status_update(ts: float, input_other: int, output: int,
 
 
 def test_parse_wire_extracts_steps(tmp_path):
-    now = datetime.now(timezone.utc).timestamp()
-    wire = _make_wire(tmp_path, [
-        _status_update(now, 1000, 200),
-        _status_update(now + 1, 2000, 400),
-    ])
+    now = datetime.now(UTC).timestamp()
+    wire = _make_wire(
+        tmp_path,
+        [
+            _status_update(now, 1000, 200),
+            _status_update(now + 1, 2000, 400),
+        ],
+    )
     session_id = "test-session"
     seen = set()
     steps = list(parse_wire_file(wire, session_id, seen))
@@ -64,7 +67,7 @@ def test_parse_wire_extracts_steps(tmp_path):
 
 
 def test_parse_wire_deduplicates(tmp_path):
-    now = datetime.now(timezone.utc).timestamp()
+    now = datetime.now(UTC).timestamp()
     # Same timestamp and total_tokens → duplicate
     record = _status_update(now, 1000, 200)
     wire = _make_wire(tmp_path, [record, record])
@@ -74,7 +77,7 @@ def test_parse_wire_deduplicates(tmp_path):
 
 
 def test_parse_wire_unknown_model_defaults(tmp_path):
-    now = datetime.now(timezone.utc).timestamp()
+    now = datetime.now(UTC).timestamp()
     wire = _make_wire(tmp_path, [_status_update(now, 500, 100)])
     seen = set()
     steps = list(parse_wire_file(wire, "s", seen))
@@ -83,11 +86,18 @@ def test_parse_wire_unknown_model_defaults(tmp_path):
 
 
 def test_compute_period_stats_today(tmp_path):
-    now = datetime.now(timezone.utc).timestamp()
+    now = datetime.now(UTC).timestamp()
     steps = [
-        StepRecord(session_id="s1", timestamp=now, model_name="claude-sonnet-4-5",
-                   provider_key="anthropic", input_other=1000, output=200,
-                   input_cache_read=0, input_cache_creation=0),
+        StepRecord(
+            session_id="s1",
+            timestamp=now,
+            model_name="claude-sonnet-4-5",
+            provider_key="anthropic",
+            input_other=1000,
+            output=200,
+            input_cache_read=0,
+            input_cache_creation=0,
+        ),
     ]
     stats = compute_period_stats(steps)
     assert stats["all_time"].total_messages == 1
@@ -96,15 +106,29 @@ def test_compute_period_stats_today(tmp_path):
 
 
 def test_compute_period_stats_excludes_old(tmp_path):
-    old_ts = datetime(2020, 1, 1, tzinfo=timezone.utc).timestamp()
-    now = datetime.now(timezone.utc).timestamp()
+    old_ts = datetime(2020, 1, 1, tzinfo=UTC).timestamp()
+    now = datetime.now(UTC).timestamp()
     steps = [
-        StepRecord(session_id="s1", timestamp=old_ts, model_name="m",
-                   provider_key="p", input_other=100, output=50,
-                   input_cache_read=0, input_cache_creation=0),
-        StepRecord(session_id="s2", timestamp=now, model_name="m",
-                   provider_key="p", input_other=200, output=100,
-                   input_cache_read=0, input_cache_creation=0),
+        StepRecord(
+            session_id="s1",
+            timestamp=old_ts,
+            model_name="m",
+            provider_key="p",
+            input_other=100,
+            output=50,
+            input_cache_read=0,
+            input_cache_creation=0,
+        ),
+        StepRecord(
+            session_id="s2",
+            timestamp=now,
+            model_name="m",
+            provider_key="p",
+            input_other=200,
+            output=100,
+            input_cache_read=0,
+            input_cache_creation=0,
+        ),
     ]
     stats = compute_period_stats(steps)
     assert stats["all_time"].total_messages == 2
@@ -134,6 +158,7 @@ def test_collect_session_files_finds_wires(tmp_path):
 
 def test_load_all_stats_returns_all_stats(tmp_path, monkeypatch):
     from pythinker_code.ui.shell.stats_collector import AllStats, load_all_stats
+
     monkeypatch.setattr(
         "pythinker_code.ui.shell.stats_collector.get_sessions_root",
         lambda: tmp_path / "sessions",