/
dsboikov
/
aiBoardRoom
Обзор
Документация
Войти
/
dsboikov
/
aiBoardRoom
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
develop
tests/test_utils.py
34 строки
950 B
Денис
Project updated
03 авг 2026, 12:39
03 авг 2026, 12:39
d7da468
Код
Авторство
О чём код?
"""Token / cost utility tests.""" from __future__ import annotations from utils.cost_calculator import estimate_cost, estimate_request_cost, format_usd from utils.token_counter import count_tokens, truncate_to_tokens def test_count_tokens_nonempty(): n = count_tokens("hello world, this is a test", model="gpt-4o") assert n > 0 def test_truncate(): text = "word " * 500 truncated = truncate_to_tokens(text, 20, model="gpt-4o") assert count_tokens(truncated, "gpt-4o") <= 25 def test_estimate_cost(): est = estimate_cost("gpt-4o", 1_000_000, 1_000_000) assert est.input_cost == 2.5 assert est.output_cost == 10.0 assert est.total_cost == 12.5 def test_estimate_request(): est = estimate_request_cost( "deepseek-chat", [{"role": "user", "content": "short"}], expected_output_tokens=100, ) assert est.total_cost >= 0 assert format_usd(est.total_cost).startswith("$")