from __future__ import annotations

from lost_cities_jax.gates import load_gates_config


def test_gates_config_loads_repair_settings():
    cfg = load_gates_config("configs/jax_ppo/gates-1-2.yaml")
    assert cfg.gate2_pass_threshold == 0.55
    assert len(cfg.exploiters) == 3
    assert cfg.repair.max_cycles == 3
    assert cfg.repair.league_template == "configs/jax_ppo/league-v1.yaml"
    assert cfg.repair.evaluation_games == 2000


def test_gates_smoke_config_uses_smoke_repair_template():
    cfg = load_gates_config("configs/jax_ppo/gates-1-2-smoke.yaml")
    assert cfg.repair.max_cycles == 1
    assert cfg.repair.league_template == "configs/jax_ppo/league-smoke.yaml"
    assert cfg.repair.guard_max_steps_rate == 1.0
