monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Let stdin_prompt image gen tests be more lightweight

Browse Source
This commit is contained in:
filipstrand 2025-06-13 00:26:43 +02:00
parent d2296ec197
commit dac08fdfdc
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/image_generation/test_stdin_prompt_integration.py
View File

@ -30,6 +30,12 @@ def test_stdin_prompt_with_actual_generation(temp_output_dir):
"1",
"--seed",
"42",
"--height",
"256",
"--width",
"256",
"-q",
"4",
"--output",
str(output_image),
"--metadata",
@ -80,6 +86,12 @@ def test_stdin_prompt_multiline_with_actual_generation(temp_output_dir):
"1",
"--seed",
"123",
"--height",
"256",
"--width",
"256",
"-q",
"4",
"--output",
str(output_image),
"--metadata",
@ -111,6 +123,12 @@ def test_empty_stdin_fails_generation(temp_output_dir):
"dev",
"--steps",
"1",
"--height",
"256",
"--width",
"256",
"-q",
"4",
"--output",
str(output_image),
]
@ -135,7 +153,7 @@ def test_pipe_from_echo_command(temp_output_dir):
metadata_file = temp_output_dir / "test_echo.json"
# Simulate: echo "prompt" | mflux-generate --prompt - ...
echo_cmd = f'echo "{prompt}" | {sys.executable} -m mflux.generate --prompt - --model dev --steps 1 --output {output_image} --metadata'
echo_cmd = f'echo "{prompt}" | {sys.executable} -m mflux.generate --prompt - --model dev --steps 1 --height 256 --width 256 -q 4 --output {output_image} --metadata'
process = subprocess.run(echo_cmd, shell=True, capture_output=True, text=True)