Function test_rerank_topk

tests/test_ranking.py:9–21

rerank_topk: empty → []; penalise_paths=False respects raw scores; saturation decay keeps order.

()

def test_rerank_topk() -> None:
    """rerank_topk: empty → []; penalise_paths=False respects raw scores; saturation decay keeps order."""
    assert rerank_topk({}, top_k=5) == []

    init_chunk = make_chunk("from .auth import authenticate", "src/semble/__init__.py")
    impl_chunk = make_chunk("def authenticate(token): ...", "src/semble/auth.py")
    ranked = rerank_topk({init_chunk: 2.0, impl_chunk: 1.0}, top_k=2, penalise_paths=False)
    assert ranked[0][0] == init_chunk

    saturated = [make_chunk(f"def fn_{i}(): pass", "big_file.py") for i in range(5)]
    ranked_sat = rerank_topk({c: float(5 - i) for i, c in enumerate(saturated)}, top_k=5)
    scores = [s for _, s in ranked_sat]
    assert scores == sorted(scores, reverse=True)


@pytest.mark.parametrize(