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

11
tests/arg_parser/test_cli_argparser.py
View File

@ -392,6 +392,8 @@ 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
# Mock get_lora_path to bypass file validation for test purposes
with patch("mflux.ui.cli.parsers.get_lora_path", side_effect=lambda x: x):
# test metadata config accepted # test metadata config accepted
with patch('sys.argv', mflux_generate_minimal_argv + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off with patch('sys.argv', mflux_generate_minimal_argv + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off
args = mflux_generate_parser.parse_args() args = mflux_generate_parser.parse_args()
@ -399,7 +401,14 @@ def test_lora_args(mflux_generate_parser, mflux_generate_minimal_argv, base_meta
assert args.lora_scales == [pytest.approx(0.3), pytest.approx(0.7)] 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 = [
"--lora-paths",
"/some/lora/3.safetensors",
"/some/lora/4.safetensors",
"--lora-scales",
"0.1",
"0.9",
]
with patch('sys.argv', mflux_generate_minimal_argv + new_loras + ['--config-from-metadata', metadata_file.as_posix()]): # fmt: off 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() args = mflux_generate_parser.parse_args()
assert len(args.lora_paths) == 4 assert len(args.lora_paths) == 4