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

Fix test for test_lora_args

Browse Source
This commit is contained in:
filipstrand 2025-06-04 21:47:26 +02:00
parent 13bfb24964
commit 674c44e30d
1 changed files with 22 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/arg_parser/test_cli_argparser.py
View File

@ -392,20 +392,29 @@ def test_lora_args(mflux_generate_parser, mflux_generate_minimal_argv, base_meta
assert args.lora_paths is None assert args.lora_paths is None
assert args.lora_scales is None assert args.lora_scales is None
# test metadata config accepted # Mock get_lora_path to bypass file validation for test purposes
with patch('sys.argv', mflux_generate_minimal_argv + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off with patch("mflux.ui.cli.parsers.get_lora_path", side_effect=lambda x: x):
args = mflux_generate_parser.parse_args() # test metadata config accepted
assert args.lora_paths == test_paths with patch('sys.argv', mflux_generate_minimal_argv + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off
assert args.lora_scales == [pytest.approx(0.3), pytest.approx(0.7)] args = mflux_generate_parser.parse_args()
assert args.lora_paths == test_paths
assert args.lora_scales == [pytest.approx(0.3), pytest.approx(0.7)]
# test CLI override that merges CLI loras and config file loras # test CLI override that merges CLI loras and config file loras
new_loras = ["--lora-paths", "/some/lora/3.safetensors", "/some/lora/4.safetensors", "--lora-scales", "0.1", "0.9"] new_loras = [
with patch('sys.argv', mflux_generate_minimal_argv + new_loras + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off "--lora-paths",
args = mflux_generate_parser.parse_args() "/some/lora/3.safetensors",
assert len(args.lora_paths) == 4 "/some/lora/4.safetensors",
assert args.lora_paths == test_paths + new_loras[1:3] "--lora-scales",
assert len(args.lora_scales) == 4 "0.1",
assert args.lora_scales == [pytest.approx(v) for v in [0.3, 0.7, 0.1, 0.9]] "0.9",
]
with patch('sys.argv', mflux_generate_minimal_argv + new_loras + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off
args = mflux_generate_parser.parse_args()
assert len(args.lora_paths) == 4
assert args.lora_paths == test_paths + new_loras[1:3]
assert len(args.lora_scales) == 4
assert args.lora_scales == [pytest.approx(v) for v in [0.3, 0.7, 0.1, 0.9]]
def test_image_to_image_args(mflux_generate_parser, mflux_generate_minimal_argv, base_metadata_dict, temp_dir): # fmt: off def test_image_to_image_args(mflux_generate_parser, mflux_generate_minimal_argv, base_metadata_dict, temp_dir): # fmt: off