let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Qing
2023-03-27 20:54:05 +08:00
parent 2e6e87275e
commit 8af5df491e
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lama_cleaner/tests/test_model.py
View File

@@ -132,16 +132,20 @@ def test_zits(strategy, zits_wireframe):
) )
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL], "no_half", [True, False]) @pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
@pytest.mark.parametrize("no_half", [True, False])
def test_mat(strategy, no_half): def test_mat(strategy, no_half):
model = ModelManager(name="mat", device=device, no_half=no_half) model = ModelManager(name="mat", device=device, no_half=no_half)
cfg = get_config(strategy) cfg = get_config(strategy)
for _ in range(10): for _ in range(10):
assert_equal( assert_equal(
model, cfg, f"mat_{strategy.capitalize()}_result.png", model,
cfg,
f"mat_{strategy.capitalize()}_result.png",
) )
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL]) @pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
def test_fcf(strategy): def test_fcf(strategy):
model = ModelManager(name="fcf", device=device) model = ModelManager(name="fcf", device=device)
@@ -188,4 +192,3 @@ def test_manga(strategy):
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png", img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
) )
>>>>>>> 944494a (mat support float16)