20 lines
731 B
Python
20 lines
731 B
Python
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
|