From dda88a6f86447ff0168717b7328d0bf350c17df0 Mon Sep 17 00:00:00 2001 From: wangyu <53896905+yenuo26@users.noreply.github.com> Date: Sun, 12 Jul 2026 17:50:54 +0800 Subject: [PATCH] [BugFix] Align invalid layers DFX tests with 2-10 range (#5044) (#5047) Signed-off-by: wangyu <410167048@qq.com> --- .../invalid_param_test/test_invalid_image_editing.py | 4 ++-- .../invalid_param_test/test_invalid_image_generation.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/dfx/reliability/invalid_param_test/test_invalid_image_editing.py b/tests/dfx/reliability/invalid_param_test/test_invalid_image_editing.py index fa7f0f905..5ffd25632 100644 --- a/tests/dfx/reliability/invalid_param_test/test_invalid_image_editing.py +++ b/tests/dfx/reliability/invalid_param_test/test_invalid_image_editing.py @@ -62,8 +62,8 @@ def _finalize_edits_form_data(template: dict[str, Any], server_model: str) -> di ), pytest.param( True, - {"prompt": "edit", "model": _IMAGE_EDITS_SERVER_MODEL, "layers": "2"}, - ("Invalid layers", "3", "1"), + {"prompt": "edit", "model": _IMAGE_EDITS_SERVER_MODEL, "layers": "1"}, + ("Invalid layers", "2", "10"), 400, id="invalid_layers", ), diff --git a/tests/dfx/reliability/invalid_param_test/test_invalid_image_generation.py b/tests/dfx/reliability/invalid_param_test/test_invalid_image_generation.py index 6f7cf984d..5e9424c88 100644 --- a/tests/dfx/reliability/invalid_param_test/test_invalid_image_generation.py +++ b/tests/dfx/reliability/invalid_param_test/test_invalid_image_generation.py @@ -87,8 +87,8 @@ def _minimal_images_gen_json(omni_server: OmniServer) -> dict[str, object]: pytest.param( {"true_cfg_scale": 21.0}, ("true_cfg_scale", "less_than_equal", "20"), id="true_cfg_scale_above_max" ), - pytest.param({"layers": 2}, ("layers", "value_error", "3", "10"), id="layers_below_min"), - pytest.param({"layers": 11}, ("layers", "value_error", "3", "10"), id="layers_above_max"), + pytest.param({"layers": 1}, ("layers", "value_error", "2", "10"), id="layers_below_min"), + pytest.param({"layers": 11}, ("layers", "value_error", "2", "10"), id="layers_above_max"), pytest.param({"seed": -1}, ("seed", "greater_than_equal", "0"), id="seed_negative", marks=_SKIP_ISSUE_3649), pytest.param( {"seed": 2**32},