diff --git a/lama_cleaner/tests/overture-creations-5sI6fQgYIuo_mask_blur.png b/lama_cleaner/tests/overture-creations-5sI6fQgYIuo_mask_blur.png new file mode 100644 index 0000000..a630379 Binary files /dev/null and b/lama_cleaner/tests/overture-creations-5sI6fQgYIuo_mask_blur.png differ diff --git a/lama_cleaner/tests/test_model.py b/lama_cleaner/tests/test_model.py index dadcbc9..accdae3 100644 --- a/lama_cleaner/tests/test_model.py +++ b/lama_cleaner/tests/test_model.py @@ -176,6 +176,14 @@ def test_sd(strategy, sampler, capfd): mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", ) + assert_equal( + model, + cfg, + f"sd_{strategy.capitalize()}_{sampler}_blur_mask_result.png", + img_p=current_dir / "overture-creations-5sI6fQgYIuo.png", + mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask_blur.png", + ) + # captured = capfd.readouterr() # for i in range(sd_steps): # assert f'sd_step_{i}' in captured.out