Jesse_Chen
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
from scripts.minute_rectification_fact_ranker_v4 import (
|
|
ALGORITHM_VERSION,
|
|
LAYER_FAMILIES,
|
|
rank_fact_rows,
|
|
score_fact_ranker_v4,
|
|
)
|
|
|
|
|
|
def _fact(event_id: str, domain: str, *, varga_house: int, av: int) -> dict:
|
|
return {
|
|
"event_id": event_id,
|
|
"domain": domain,
|
|
"vimshottari": {"md": "Sun", "ad": "Moon", "pd": "Mars"},
|
|
"narayana": {"md_sign": 0, "ad_sign": 1},
|
|
"d1": {
|
|
"ascendant_sign": 0,
|
|
"target_house_lords": ["Moon"],
|
|
"active_lord_houses": {"md": 1, "ad": 7, "pd": 3},
|
|
},
|
|
"vargas": [{
|
|
"chart": "D9" if domain == "relationship" else "D10",
|
|
"ascendant_sign": 0,
|
|
"active_lord_houses": {"md": varga_house, "ad": 3, "pd": 4},
|
|
}],
|
|
"arudha_signs": {"A7": 0} if domain == "relationship" else {"A10": 0},
|
|
"ashtakavarga_target_house_scores": {"7": av},
|
|
"verified_shadbala_state": ["shadbala_sthana_drik_naisargika_support_auxiliary"],
|
|
}
|
|
|
|
|
|
def _rows() -> list[dict]:
|
|
return [
|
|
{
|
|
"time": candidate,
|
|
"feature_contract_version": "minute-rectification-feature-facts-v4-shadow",
|
|
"event_facts": [
|
|
_fact("event-career", "career", varga_house=varga_house, av=av),
|
|
_fact("event-relationship", "relationship", varga_house=varga_house, av=av),
|
|
],
|
|
"missing_layers": [],
|
|
}
|
|
for candidate, varga_house, av in (
|
|
("10:00", 3, 20),
|
|
("10:01", 10, 32),
|
|
("10:02", 3, 20),
|
|
)
|
|
]
|
|
|
|
|
|
def _events() -> list[dict]:
|
|
return [
|
|
{"id": "event-career", "domain": "career", "date": "2020", "precision": "year"},
|
|
{"id": "event-relationship", "domain": "relationship", "date": "2021-03", "precision": "month"},
|
|
]
|
|
|
|
|
|
def test_fact_ranker_uses_candidate_differences_and_balances_domains() -> None:
|
|
rows, contract = rank_fact_rows(_rows(), _events())
|
|
|
|
assert max(rows, key=lambda row: row["score"])["time"] == "10:01"
|
|
assert contract["algorithm_version"] == ALGORITHM_VERSION
|
|
assert contract["shadow_only"] is True
|
|
assert set(contract["layer_weights"]) == set(LAYER_FAMILIES)
|
|
assert set(contract["candidate_domain_scores"]["10:01"]) == {"career", "relationship"}
|
|
|
|
|
|
def test_fact_ranker_ablation_and_safety_gates_never_apply_a_minute() -> None:
|
|
result = score_fact_ranker_v4(_rows(), _events())
|
|
|
|
assert result["can_apply"] is False
|
|
assert result["shadow_only"] is True
|
|
assert "fact_ranker_v4_holdout_not_ready" in result["reasons"]
|
|
assert {
|
|
run["removed_layer"] for run in result["stability_diagnostics"]["ablation"]["runs"]
|
|
} == set(LAYER_FAMILIES)
|
|
assert result["stability_diagnostics"]["neighbor_stability"]["all_required_passed"] is False
|
|
assert result["stability_diagnostics"]["leave_one_event_out"]["status"] == "pass"
|
|
|
|
|
|
def test_constant_fact_layers_contribute_zero_candidate_preference() -> None:
|
|
rows = _rows()
|
|
for row in rows:
|
|
for fact in row["event_facts"]:
|
|
fact["vargas"][0]["active_lord_houses"]["md"] = 3
|
|
fact["ashtakavarga_target_house_scores"]["7"] = 28
|
|
|
|
ranked, contract = rank_fact_rows(rows, _events())
|
|
|
|
assert len({row["score"] for row in ranked}) == 1
|
|
assert all(score == 0 for score in contract["event_raw_scores"]["event-career"].values())
|