from pathlib import Path


def test_retrain_runner_passes_training_horizon_to_calibrator() -> None:
    runner = (
        Path(__file__).resolve().parents[1] / "tools" / "run_torch_retrain.ps1"
    ).read_text(encoding="utf-8")

    calibration_start = runner.index("$calibrationBaseArgs = @(")
    calibration_end = runner.index("\n    )", calibration_start)
    calibration_args = runner[calibration_start:calibration_end]

    assert '"--horizon", $Horizon.ToString()' in calibration_args


def test_retrain_runner_uses_a_regime_sized_validation_window() -> None:
    runner = (
        Path(__file__).resolve().parents[1] / "tools" / "run_torch_retrain.ps1"
    ).read_text(encoding="utf-8")

    assert "[int]$ValidationWindow = 0" in runner
    assert "else { 720 }" in runner
    assert '"--validation-window", $ValidationWindow.ToString()' in runner