24 lines
859 B
Python
24 lines
859 B
Python
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
|