145 lines
5.1 KiB
Python
145 lines
5.1 KiB
Python
"""Cost engine — including the spec's acceptance numbers."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pytest
|
||
|
||
from tokencalc.cost_model import (
|
||
calculate_consumption_ai_cost,
|
||
calculate_per_user_ai_cost,
|
||
calculate_platform_license_cost,
|
||
calculate_total_cost,
|
||
)
|
||
from tokencalc.defaults import (
|
||
CONTRACTED_NAMED_USERS,
|
||
CTM_DEFAULT_FEATURE_SCOPES,
|
||
CTM_DEFAULT_SITES,
|
||
DEFAULT_METERS,
|
||
DEFAULT_PRICING,
|
||
)
|
||
from tokencalc.inputs import FeatureScope, SiteInput
|
||
from tokencalc.scenarios import get_scenario
|
||
|
||
ALL_SITES = [s.site_name for s in CTM_DEFAULT_SITES]
|
||
|
||
|
||
def _scope(feature, sites=None, **kw):
|
||
return FeatureScope(feature, sites or ALL_SITES, **kw)
|
||
|
||
|
||
def test_default_sites_match_contracted_users():
|
||
assert sum(s.named_users for s in CTM_DEFAULT_SITES) == CONTRACTED_NAMED_USERS
|
||
|
||
|
||
def test_sta_acceptance_number():
|
||
"""2,088 users × 30 tokens × 12 months × $1 = $751,680."""
|
||
df = calculate_per_user_ai_cost(
|
||
CTM_DEFAULT_SITES, _scope("Speech & Text Analytics [named]"),
|
||
DEFAULT_METERS["Speech & Text Analytics [named]"], DEFAULT_PRICING,
|
||
)
|
||
assert df["annual_cost"].sum() == pytest.approx(751_680)
|
||
|
||
|
||
def test_agent_copilot_acceptance_number():
|
||
"""2,088 users × 40 tokens × 12 months × $1 = $1,002,240."""
|
||
df = calculate_per_user_ai_cost(
|
||
CTM_DEFAULT_SITES, _scope("Agent Copilot [named]"),
|
||
DEFAULT_METERS["Agent Copilot [named]"], DEFAULT_PRICING,
|
||
)
|
||
assert df["annual_cost"].sum() == pytest.approx(1_002_240)
|
||
|
||
|
||
def test_ai_translate_not_active_before_phase():
|
||
"""AI Translate (consumption meter) produces zero cost before its phase."""
|
||
scenario = get_scenario("realistic")
|
||
apac_sites = [s.site_name for s in CTM_DEFAULT_SITES if s.region_pricing == "APAC"]
|
||
df = calculate_consumption_ai_cost(
|
||
CTM_DEFAULT_SITES,
|
||
_scope("AI Translate", apac_sites, phase=3),
|
||
DEFAULT_METERS["AI Translate"], scenario, DEFAULT_PRICING, year=2,
|
||
)
|
||
assert df["annual_cost"].sum() == 0
|
||
|
||
|
||
def test_copilot_covers_supervisor_summary():
|
||
"""Rule 1: AI Summary cost is zero at Copilot sites."""
|
||
scenario = get_scenario("realistic")
|
||
total = calculate_total_cost(
|
||
CTM_DEFAULT_SITES,
|
||
[
|
||
_scope("Agent Copilot [named]"),
|
||
_scope("AI Summary & Insights"),
|
||
],
|
||
DEFAULT_METERS, DEFAULT_PRICING, scenario, year=1,
|
||
include_platform=False,
|
||
)
|
||
summary_row = total[total["cost_line"] == "AI Summary & Insights"].iloc[0]
|
||
assert summary_row["annual_cost"] == 0
|
||
# Without Copilot the same line costs real money.
|
||
total2 = calculate_total_cost(
|
||
CTM_DEFAULT_SITES,
|
||
[_scope("AI Summary & Insights")],
|
||
DEFAULT_METERS, DEFAULT_PRICING, scenario, year=1,
|
||
include_platform=False,
|
||
)
|
||
assert total2[total2["cost_line"] == "AI Summary & Insights"].iloc[0][
|
||
"annual_cost"
|
||
] > 0
|
||
|
||
|
||
def test_consumption_tokens_rounded_up_monthly():
|
||
"""Rule 2: ceil on monthly site token totals."""
|
||
site = SiteInput(
|
||
"Tiny", "US", agents=5, supervisors=0,
|
||
voice_volume_monthly=100, email_volume_monthly=0,
|
||
chat_volume_monthly=0, sms_volume_monthly=0,
|
||
voice_aht_seconds=300, email_aht_seconds=600,
|
||
chat_aht_seconds=480, voice_acw_seconds=60,
|
||
fully_loaded_agent_cost_annual=65_000,
|
||
fully_loaded_supervisor_cost_annual=95_000,
|
||
)
|
||
# realistic: 100 calls × 35% × 1.5 min = 52.5 min × (1/17) = 3.088
|
||
# tokens × 70% Y1 ramp applied to units → 36.75 min → 2.16 tokens → ceil 3
|
||
df = calculate_consumption_ai_cost(
|
||
[site], FeatureScope("Voice Bot", ["Tiny"]),
|
||
DEFAULT_METERS["Voice Bot"], "realistic", DEFAULT_PRICING, year=1,
|
||
)
|
||
assert df.iloc[0]["tokens_monthly"] == 3
|
||
assert df.iloc[0]["annual_cost"] == pytest.approx(3 * 12 * 1.0)
|
||
|
||
|
||
def test_regional_pricing_not_hardcoded():
|
||
pricing = dict(DEFAULT_PRICING)
|
||
from tokencalc.meters import TokenPricing
|
||
|
||
pricing["APAC"] = TokenPricing(region="APAC", list_rate_per_token=2.0)
|
||
apac_site = next(s for s in CTM_DEFAULT_SITES if s.region_pricing == "APAC")
|
||
df = calculate_per_user_ai_cost(
|
||
[apac_site], _scope("Speech & Text Analytics [named]", [apac_site.site_name]),
|
||
DEFAULT_METERS["Speech & Text Analytics [named]"], pricing,
|
||
)
|
||
expected = apac_site.named_users * 30 * 12 * 2.0
|
||
assert df["annual_cost"].sum() == pytest.approx(expected)
|
||
|
||
|
||
def test_year1_consumption_ramp_default_70pct():
|
||
sc = get_scenario("realistic")
|
||
assert sc.cost_realization(1) == pytest.approx(0.70)
|
||
assert sc.cost_realization(2) == 1.0
|
||
|
||
|
||
def test_platform_license_cost():
|
||
df = calculate_platform_license_cost(CTM_DEFAULT_SITES)
|
||
expected = CONTRACTED_NAMED_USERS * 111.28 * 12
|
||
assert df["annual_cost"].sum() == pytest.approx(expected)
|
||
|
||
|
||
def test_total_cost_default_scopes_runs_all_years():
|
||
for year in (1, 2, 3):
|
||
df = calculate_total_cost(
|
||
CTM_DEFAULT_SITES, CTM_DEFAULT_FEATURE_SCOPES,
|
||
DEFAULT_METERS, DEFAULT_PRICING, "realistic", year,
|
||
)
|
||
assert (df["annual_cost"] >= 0).all()
|
||
assert {"cost_line", "scope", "annual_cost", "confidence"} <= set(df.columns)
|