test: add safety guards and unit tests for video loop calculators

Added input validations (`source_frame_count <= 0` or `target_frames_per_loop <= 0`) to `AutoLoopCalculatorLTX` (in `loop.py`), and `AutoLoopCalculator`, `AutoLoopCalculatorWan` (in `video.py`). When conditions are met, a safe default is returned to prevent math domain errors. Also included a new comprehensive unit test suite in `tests/test_loop_calculator.py`.
This commit is contained in:
Meisoftcoltd
2026-05-14 20:24:55 +00:00
parent 9df7a16940
commit 3afcee2bef
3 changed files with 158 additions and 0 deletions
+6
View File
@@ -95,6 +95,12 @@ class AutoLoopCalculatorLTX:
CATEGORY = "🔁 Sequential Batcher/Loop"
def calculate(self, source_frame_count, target_frames_per_loop, select_every_nth, current_loop_index, safe_faces_list=None, scene_cuts_list=None):
if source_frame_count <= 0 or target_frames_per_loop <= 0:
return (1, 0, select_every_nth)
source_frame_count = min(source_frame_count, 10000)
select_every_nth = max(1, select_every_nth)
# --- Ajuste Proporcional LTX (Regla DiT 8n + 1) ---
import math
from . import loop