"""Unit tests for V38 live-conditioned projection (pure, no DB/model deps).""" import os import sys sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) from models.live_matrix import ( build_live_projection, estimate_minute, state_multiplier, ) def _approx(a, b, tol=1e-6): return abs(a - b) <= tol def test_probs_form_distribution(): proj = build_live_projection(0.50, 0.27, 0.23, 0.55, 1, 0, 60) p = proj["probs"] assert _approx(p["1"] + p["X"] + p["2"], 1.0, 1e-3) assert 0.0 <= proj["p_away_scores_again"] <= 1.0 def test_minute_one_roughly_matches_prematch(): # at 0-0 minute 1 the projection must stay close to the anchored numbers proj = build_live_projection(0.50, 0.27, 0.23, 0.55, 0, 0, 1) assert abs(proj["probs"]["1"] - 0.50) < 0.06 assert abs(proj["probs"]["2"] - 0.23) < 0.06 def test_one_goal_lead_at_80(): # the user's exact case: 1-0 at 80' (OOS-validated: said 21.7 / actual 23.0) proj = build_live_projection(0.50, 0.27, 0.23, 0.55, 1, 0, 80) assert proj["probs"]["1"] > 0.72 # leader is now strong fav assert 0.08 <= proj["p_away_scores_again"] <= 0.30 assert _approx( proj["p_comeback"], proj["probs"]["X"] + proj["probs"]["2"], 1e-9 ) def test_less_time_means_fewer_chances(): early = build_live_projection(0.50, 0.27, 0.23, 0.55, 1, 0, 60) late = build_live_projection(0.50, 0.27, 0.23, 0.55, 1, 0, 85) assert late["p_away_scores_again"] < early["p_away_scores_again"] assert late["probs"]["1"] > early["probs"]["1"] def test_trailing_team_pushes_late(): assert state_multiplier(-1, 80) > 1.05 # trailing by one, late: pushes assert state_multiplier(1, 80) < 1.0 # leading by one, late: parks bus assert state_multiplier(-1, 80) > state_multiplier(-1, 30) def test_score_consistency_with_current_score(): proj = build_live_projection(0.50, 0.27, 0.23, 0.55, 2, 1, 75) # every scenario must be reachable from the current score for s in proj["scenario_top5"]: fh, fa = map(int, str(s["score"]).split("-")) assert fh >= 2 and fa >= 1 assert proj["current_score"] == "2-1" def test_estimate_minute_approximation(): now = 1_700_000_000_000 assert estimate_minute(None, now) is None assert estimate_minute(now + 60_000, now) is None # not kicked off assert estimate_minute(now - 30 * 60_000, now) == 30 # mid 1H assert estimate_minute(now - 55 * 60_000, now) == 46 # HT break assert estimate_minute(now - 80 * 60_000, now) == 65 # 2H, break folded assert estimate_minute(now - 200 * 60_000, now) == 94 # capped if __name__ == "__main__": fns = [v for k, v in sorted(globals().items()) if k.startswith("test_")] for fn in fns: fn() print(f"PASS {fn.__name__}") print(f"\nAll {len(fns)} tests passed.")