16 lines
425 B
Python
16 lines
425 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():
|
|
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
|