mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-06 13:57:08 +08:00
11a35d6dfb
Co-authored-by: comfyanonymous <comfyanonymous@protonmail.com> Co-authored-by: AustinMroz <austinmroz@utexas.edu> Co-authored-by: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Co-authored-by: Benjamin Lu <benceruleanlu@proton.me> Co-authored-by: Andrew Kvochko <kvochko@users.noreply.github.com> Co-authored-by: Pam <42671363+pamparamm@users.noreply.github.com> Co-authored-by: chaObserv <154517000+chaObserv@users.noreply.github.com> Co-authored-by: Yoland Yan <4950057+yoland68@users.noreply.github.com> Co-authored-by: guill <guill@users.noreply.github.com> Co-authored-by: Chenlei Hu <hcl@comfy.org> Co-authored-by: Terry Jia <terryjia88@gmail.com> Co-authored-by: Silver <65376327+silveroxides@users.noreply.github.com> Co-authored-by: Christian Byrne <cbyrne@comfy.org> Co-authored-by: catboxanon <122327233+catboxanon@users.noreply.github.com> Co-authored-by: liesen <liesen.dev@gmail.com> Co-authored-by: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> Co-authored-by: Robin Huang <robin.j.huang@gmail.com> Co-authored-by: thot-experiment <thot@thiic.cc> Co-authored-by: thot experiment <94414189+thot-experiment@users.noreply.github.com>
92 lines
3.5 KiB
Python
92 lines
3.5 KiB
Python
import io
|
|
from comfy_api.input_impl.video_types import (
|
|
container_to_output_format,
|
|
get_open_write_kwargs,
|
|
)
|
|
from comfy_api.util import VideoContainer
|
|
|
|
|
|
def test_container_to_output_format_empty_string():
|
|
"""Test that an empty string input returns None. `None` arg allows default auto-detection."""
|
|
assert container_to_output_format("") is None
|
|
|
|
|
|
def test_container_to_output_format_none():
|
|
"""Test that None input returns None."""
|
|
assert container_to_output_format(None) is None
|
|
|
|
|
|
def test_container_to_output_format_comma_separated():
|
|
"""Test that a comma-separated list returns a valid singular format from the list."""
|
|
comma_separated_format = "mp4,mov,m4a"
|
|
output_format = container_to_output_format(comma_separated_format)
|
|
assert output_format in comma_separated_format
|
|
|
|
|
|
def test_container_to_output_format_single():
|
|
"""Test that a single format string (not comma-separated list) is returned as is."""
|
|
assert container_to_output_format("mp4") == "mp4"
|
|
|
|
|
|
def test_get_open_write_kwargs_filepath_no_format():
|
|
"""Test that 'format' kwarg is NOT set when dest is a file path."""
|
|
kwargs_auto = get_open_write_kwargs("output.mp4", "mp4", VideoContainer.AUTO)
|
|
assert "format" not in kwargs_auto, "Format should not be set for file paths (AUTO)"
|
|
|
|
kwargs_specific = get_open_write_kwargs("output.avi", "mp4", "avi")
|
|
fail_msg = "Format should not be set for file paths (Specific)"
|
|
assert "format" not in kwargs_specific, fail_msg
|
|
|
|
|
|
def test_get_open_write_kwargs_base_options_mode():
|
|
"""Test basic kwargs for file path: mode and movflags."""
|
|
kwargs = get_open_write_kwargs("output.mp4", "mp4", VideoContainer.AUTO)
|
|
assert kwargs["mode"] == "w", "mode should be set to write"
|
|
|
|
fail_msg = "movflags should be set to preserve custom metadata tags"
|
|
assert "movflags" in kwargs["options"], fail_msg
|
|
assert kwargs["options"]["movflags"] == "use_metadata_tags", fail_msg
|
|
|
|
|
|
def test_get_open_write_kwargs_bytesio_auto_format():
|
|
"""Test kwargs for BytesIO dest with AUTO format."""
|
|
dest = io.BytesIO()
|
|
container_fmt = "mov,mp4,m4a"
|
|
kwargs = get_open_write_kwargs(dest, container_fmt, VideoContainer.AUTO)
|
|
|
|
assert kwargs["mode"] == "w"
|
|
assert kwargs["options"]["movflags"] == "use_metadata_tags"
|
|
|
|
fail_msg = (
|
|
"Format should be a valid format from the container's format list when AUTO"
|
|
)
|
|
assert kwargs["format"] in container_fmt, fail_msg
|
|
|
|
|
|
def test_get_open_write_kwargs_bytesio_specific_format():
|
|
"""Test kwargs for BytesIO dest with a specific single format."""
|
|
dest = io.BytesIO()
|
|
container_fmt = "avi"
|
|
to_fmt = VideoContainer.MP4
|
|
kwargs = get_open_write_kwargs(dest, container_fmt, to_fmt)
|
|
|
|
assert kwargs["mode"] == "w"
|
|
assert kwargs["options"]["movflags"] == "use_metadata_tags"
|
|
|
|
fail_msg = "Format should be the specified format (lowercased) when output format is not AUTO"
|
|
assert kwargs["format"] == "mp4", fail_msg
|
|
|
|
|
|
def test_get_open_write_kwargs_bytesio_specific_format_list():
|
|
"""Test kwargs for BytesIO dest with a specific comma-separated format."""
|
|
dest = io.BytesIO()
|
|
container_fmt = "avi"
|
|
to_fmt = "mov,mp4,m4a" # A format string that is a list
|
|
kwargs = get_open_write_kwargs(dest, container_fmt, to_fmt)
|
|
|
|
assert kwargs["mode"] == "w"
|
|
assert kwargs["options"]["movflags"] == "use_metadata_tags"
|
|
|
|
fail_msg = "Format should be a valid format from the specified format list when output format is not AUTO"
|
|
assert kwargs["format"] in to_fmt, fail_msg
|