Meisoftcoltd
cf6666885a
- Added global tracking variables (`global_batch_index`, `global_has_more_batches`, `global_is_batch_advancing`) to `loop.py` to maintain batch state - Updated `SequentialLoopStart` reset logic to prevent loop index reset on automatic batch advances - Implemented the `AutoLoopCalculatorTTSBatch` node with `INPUT_IS_LIST = True` returning list-wrapped chunks to bypass native ComfyUI parallelization - Updated `SequentialLoopTrigger` requeue logic to handle batch completion correctly using `reset_loop=True` with `loop_idx=0` when `has_more_batches` is truthy
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import sys
|
|
import types
|
|
import pytest
|
|
from unittest.mock import MagicMock
|
|
import os
|
|
|
|
# Mock dependencies
|
|
mock_torch = types.ModuleType("torch")
|
|
class MockTensor:
|
|
pass
|
|
mock_torch.Tensor = MockTensor
|
|
mock_torch.Tensor.__module__ = "torch"
|
|
sys.modules["torch"] = mock_torch
|
|
|
|
def get_VAESafeFramePadder():
|
|
tools_path = os.path.join(os.path.dirname(__file__), "..", "tools.py")
|
|
with open(tools_path, 'r') as f:
|
|
code = f.read()
|
|
|
|
# Mock 'from . import register_node' and other potential relative imports
|
|
code = code.replace("from . import register_node", "def register_node(c, *args, **kwargs): return c")
|
|
code = code.replace("from .switch_node import any_type", "any_type = '*'")
|
|
|
|
tools_module = types.ModuleType("tools")
|
|
# Provide torch to the module's namespace
|
|
tools_module.__dict__["torch"] = mock_torch
|
|
|
|
try:
|
|
exec(code, tools_module.__dict__)
|
|
except Exception as e:
|
|
print(f"Error executing tools.py code: {e}")
|
|
raise e
|
|
|
|
return tools_module.VAESafeFramePadder
|
|
|
|
@pytest.fixture
|
|
def VAESafeFramePadder():
|
|
return get_VAESafeFramePadder()
|
|
|
|
def test_pad_non_tensor_input_raises_value_error(VAESafeFramePadder):
|
|
padder = VAESafeFramePadder()
|
|
|
|
with pytest.raises(ValueError, match=r"\[VAESafeFramePadder\] Input is not a valid tensor\."):
|
|
padder.pad(None, "WanVideo")
|
|
|
|
with pytest.raises(ValueError, match=r"\[VAESafeFramePadder\] Input is not a valid tensor\."):
|
|
padder.pad("not a tensor", "WanVideo")
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__])
|