From dac08fdfdc3402059c6d02b19ae2a4708dee0aba Mon Sep 17 00:00:00 2001 From: filipstrand Date: Fri, 13 Jun 2025 00:26:43 +0200 Subject: [PATCH] Let stdin_prompt image gen tests be more lightweight --- .../test_stdin_prompt_integration.py | 20 ++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) diff --git a/tests/image_generation/test_stdin_prompt_integration.py b/tests/image_generation/test_stdin_prompt_integration.py index e7d4322..1867fc7 100644 --- a/tests/image_generation/test_stdin_prompt_integration.py +++ b/tests/image_generation/test_stdin_prompt_integration.py @@ -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)