Ed_
60396f03f8
Applied 236 return type annotations to functions with no return values across 100+ files (core modules, tests, scripts, simulations). Added Phase 4 to python_style_refactor track for remaining 597 items (untyped params, vars, and functions with return values). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
16 lines
418 B
Python
16 lines
418 B
Python
import pytest
|
|
import sys
|
|
import os
|
|
|
|
# Ensure project root is in path
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
import ai_client
|
|
|
|
def test_token_usage_tracking() -> None:
|
|
ai_client.reset_session()
|
|
# Mock an API response with token usage
|
|
usage = {"prompt_tokens": 100, "candidates_tokens": 50, "total_tokens": 150}
|
|
# This would test the internal accumulator in ai_client
|
|
pass
|