Claude
97b85a3f09
Shared video stream and NVIDIA helpers move to core (video_pipe, nvidia); the add-on needs nvidia-vfx 0.2+, moves an older one up and never moves a newer one back; Upscale and Interpolate cannot feed each other in Always mode. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Tw9fjW75aYB8UyVRZo6fCr
89 lines
3.3 KiB
Python
89 lines
3.3 KiB
Python
import pytest
|
|
|
|
from modules.output.render.tests.clips import comfy, make_clip, needs_ffmpeg # noqa: F401
|
|
from modules.output.render import files, upscale
|
|
|
|
torch = pytest.importorskip("torch")
|
|
|
|
|
|
@needs_ffmpeg
|
|
def test_upscale_doubles_size_keeps_fps_frames_and_sound(comfy):
|
|
src, dst = str(comfy.output / "a.mp4"), str(comfy.output / "up.mp4")
|
|
make_clip(src, seconds=1.0, size="64x48", rate=10, audio=True)
|
|
seen = []
|
|
|
|
def twice(x):
|
|
seen.append(x.shape[0])
|
|
return x.repeat_interleave(2, 1).repeat_interleave(2, 2)
|
|
|
|
assert upscale.upscale_file(src, dst, twice) == 10
|
|
assert seen == [8, 2] # streamed in chunks, not all at once
|
|
w, h, fps, sound = upscale.probe(dst)
|
|
assert (w, h, fps, sound) == (128, 96, "10/1", True)
|
|
assert files.moov_position(dst) == "front"
|
|
|
|
|
|
@needs_ffmpeg
|
|
def test_interrupt_stops_and_leaves_no_file(comfy):
|
|
src, dst = str(comfy.output / "a.mp4"), str(comfy.output / "up.mp4")
|
|
make_clip(src, seconds=1.0, size="64x48", rate=10, audio=False)
|
|
calls = []
|
|
|
|
def stop():
|
|
calls.append(1)
|
|
if len(calls) > 1:
|
|
raise KeyboardInterrupt
|
|
|
|
with pytest.raises(KeyboardInterrupt):
|
|
upscale.upscale_file(src, dst, lambda x: x, stop)
|
|
import os
|
|
assert not os.path.exists(dst)
|
|
|
|
|
|
def test_free_name_never_overwrites(tmp_path):
|
|
(tmp_path / "a_m.mp4").write_bytes(b"x")
|
|
assert upscale.free_name(str(tmp_path), "a.mp4", "sub/m.safetensors") == "a_m_1.mp4"
|
|
|
|
|
|
from modules.output.render import rtx # noqa: E402
|
|
|
|
|
|
def test_rtx_options_are_checked_by_name():
|
|
assert rtx.options("{}")["quality"] == "ULTRA"
|
|
assert rtx.options({"scale": 3, "quality": "DEBLUR_HIGH"})["scale"] == 3
|
|
for bad in ({"quality": "ULTRAA"}, {"scale": 5}, {"strength": -1}, {"resize": "huge"}, {"width": 8}, {"tenbit": "false"}):
|
|
with pytest.raises(ValueError):
|
|
rtx.options(bad)
|
|
with pytest.raises(ValueError):
|
|
rtx.options("not json")
|
|
|
|
|
|
def test_rtx_output_size():
|
|
o = rtx.options({"scale": 2})
|
|
assert rtx.out_size(640, 360, o) == (1280, 720)
|
|
assert rtx.out_size(361, 460, o) == (720, 920) # snapped to a multiple of 8
|
|
assert rtx.out_size(640, 360, rtx.options({"resize": "size", "width": 1000, "height": 510})) == (1000, 512)
|
|
assert rtx.out_size(641, 361, rtx.options({"quality": "DENOISE_HIGH", "scale": 4})) == (641, 361) # cleans at its own size
|
|
|
|
|
|
def test_rtx_ten_bit_packing_round_trips():
|
|
x = torch.rand(3, 4, 5)
|
|
back = rtx.unpack10(rtx.pack10(x))
|
|
assert back.shape == x.shape and (back - x).abs().max() < 1 / 1023
|
|
assert rtx.pack10(torch.ones(3, 1, 1)).item() == 0xFFFFFFFF # all ten-bit channels full, alpha 3
|
|
|
|
|
|
def test_rtx_without_the_add_on_says_how_to_get_it(monkeypatch):
|
|
monkeypatch.setitem(__import__("sys").modules, "nvvfx", None)
|
|
with pytest.raises(RuntimeError, match="Settings ▸ Modules"):
|
|
rtx.make(rtx.options("{}"))
|
|
|
|
|
|
def test_rtx_refuses_when_switched_off_even_if_installed(monkeypatch):
|
|
from core import addons
|
|
monkeypatch.setitem(__import__("sys").modules, "nvvfx", type("M", (), {"VideoSuperRes": object})) # the package "is installed"
|
|
monkeypatch.setattr(addons, "enabled", lambda _id: False)
|
|
with pytest.raises(RuntimeError, match="switched off"):
|
|
rtx.make(rtx.options("{}"))
|
|
|