"""对齐前端 opsSummary 规则的轻量逻辑测试（Python 镜像）。"""


MIN_N = 3
TOP_K = 5


def leverage_summary_week(buckets, min_leverage=100):
    active = [b for b in buckets if b["n"] > 0]
    ranked = sorted(
        [b for b in active if b["n"] >= MIN_N and b.get("pct_ge_min") is not None],
        key=lambda b: (b["pct_ge_min"], b["n"], b.get("mean") or 0),
        reverse=True,
    )[:TOP_K]
    return ranked


def test_week_ranks_by_pct_ge_min():
    buckets = [
        {"label": "09:00", "n": 10, "mean": 110, "pct_ge_min": 0.5},
        {"label": "14:00", "n": 10, "mean": 105, "pct_ge_min": 0.8},
        {"label": "21:00", "n": 2, "mean": 200, "pct_ge_min": 1.0},  # n 不足
    ]
    ranked = leverage_summary_week(buckets)
    assert [b["label"] for b in ranked] == ["14:00", "09:00"]


def test_day_keeps_small_n():
    active = [b for b in [
        {"label": "09:00", "n": 1, "mean": 118, "pct_ge_min": 1.0},
    ] if b["n"] > 0]
    assert len(active) == 1