67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
"""Migration + WFM (no-AI) scenario — hand-check acceptance numbers."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import pytest
|
||
|
||
from tokencalc import appendix4 as a4
|
||
from tokencalc import migration_wfm as mw
|
||
from tokencalc.defaults import CTM_DEFAULT_SITES
|
||
|
||
SITES = list(CTM_DEFAULT_SITES)
|
||
|
||
|
||
def _default_benefit_rollout():
|
||
_, _, benefit_rollout = a4.build_rollouts(SITES)
|
||
return benefit_rollout
|
||
|
||
|
||
def test_wfm_scope_and_verbatim_total():
|
||
ben = mw.wfm_benefits_by_year(_default_benefit_rollout())
|
||
assert set(ben["capability"]) == {"WFM"}
|
||
assert set(ben["region"]) == set(mw.DEFAULT_WFM_REGIONS)
|
||
assert "EMEA" not in set(ben["region"]), "EMEA WFM is out of scope"
|
||
# NA $0 (migration) + ANZ $1.4M + ASIA $914K — verbatim, exact.
|
||
assert ben["benefit"].sum() == pytest.approx(2_314_000)
|
||
|
||
|
||
def test_wfm_phasing_on_deck_schedule():
|
||
ben = mw.wfm_benefits_by_year(_default_benefit_rollout())
|
||
by_year = ben.groupby("year")["benefit"].sum()
|
||
assert by_year[2026] == 0.0
|
||
# ANZ realizes Dec 2027 (1 of 13 live months lands in 2027).
|
||
assert by_year[2027] == pytest.approx(1_400_000 / 13)
|
||
assert by_year[2028] == pytest.approx(2_314_000 - 1_400_000 / 13)
|
||
|
||
|
||
def test_runrate_saving_annual():
|
||
# (7.3M − 4.3M) licence + 1.3M ANZ + 1.6M ASIA + 0 NA = 5.9M.
|
||
assert mw.wfm_annual_runrate() == pytest.approx(2_900_000)
|
||
assert mw.runrate_saving_annual() == pytest.approx(5_900_000)
|
||
assert mw.runrate_saving_annual(regions=["NA"]) == pytest.approx(3_000_000)
|
||
assert mw.runrate_saving_annual(licence_annual=4_800_000,
|
||
regions=[]) == pytest.approx(2_500_000)
|
||
|
||
|
||
def test_breakeven_extrapolates_past_window():
|
||
cs = a4.current_state_inputs(SITES)
|
||
cur = a4.current_costs_by_year(cs)
|
||
lic = a4.licence_costs_by_year()
|
||
ps = a4.ps_costs_by_year()
|
||
total = {y: cur[y] + lic[y] + ps[y] for y in a4.YEARS}
|
||
ben = mw.wfm_benefits_by_year(_default_benefit_rollout())
|
||
ben_y = {y: float(ben.loc[ben.year == y, "benefit"].sum()) for y in a4.YEARS}
|
||
inc, net = a4.case_flows(total, ben_y)
|
||
assert sum(net.values()) == pytest.approx(-1_553_000, abs=1_000)
|
||
label = mw.runrate_breakeven_label(net, mw.runrate_saving_annual())
|
||
assert label == "40 months (~Apr 2029, extrapolated)"
|
||
|
||
|
||
def test_breakeven_defers_in_window_and_guards_zero_runrate():
|
||
positive = {2026: 1_000_000.0, 2027: 0.0, 2028: 0.0}
|
||
assert mw.runrate_breakeven_label(positive, 5_900_000) == \
|
||
a4.payback_label(positive)
|
||
negative = {2026: -1_000_000.0, 2027: 0.0, 2028: 0.0}
|
||
assert mw.runrate_breakeven_label(negative, 0.0) == \
|
||
"never at current run-rate"
|