Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-30 11:11:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(scripts): update pr options

Browse Source
This commit is contained in:
Dowon
2024-08-24 13:34:20 +09:00
parent a3506f3222
commit d705c6e30a
4 changed files with 71 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_opts.py
View File

@@ -5,7 +5,7 @@ import pytest
from hypothesis import assume, given
from hypothesis import strategies as st
from adetailer.opts import get_dynamic_denoise_strength, optimal_crop_size
from adetailer.opts import dynamic_denoise_strength, optimal_crop_size
@pytest.mark.parametrize(
@@ -17,29 +17,29 @@ from adetailer.opts import get_dynamic_denoise_strength, optimal_crop_size
(-0.5, 0.5, [0, 0, 100, 100], (1000, 1000), 0.502518907629606),
],
)
def test_get_dynamic_denoise_strength(
def test_dynamic_denoise_strength(
denoise_power: float,
denoise_strength: float,
bbox: list[int],
image_size: tuple[int, int],
expected_result: float,
):
result = get_dynamic_denoise_strength(
denoise_power, denoise_strength, bbox, image_size
)
result = dynamic_denoise_strength(denoise_power, denoise_strength, bbox, image_size)
assert np.isclose(result, expected_result)
@given(denoise_strength=st.floats(allow_nan=False))
def test_get_dynamic_denoise_strength_no_bbox(denoise_strength: float):
result = get_dynamic_denoise_strength(0.5, denoise_strength, [], (1000, 1000))
assert result == denoise_strength
def test_dynamic_denoise_strength_no_bbox(denoise_strength: float):
with pytest.raises(ValueError, match="bbox length must be 4, got 0"):
dynamic_denoise_strength(0.5, denoise_strength, [], (1000, 1000))
@given(denoise_strength=st.floats(allow_nan=False))
def test_get_dynamic_denoise_strength_zero_power(denoise_strength: float):
result = get_dynamic_denoise_strength(0.0, denoise_strength, [], (1000, 1000))
assert result == denoise_strength
def test_dynamic_denoise_strength_zero_power(denoise_strength: float):
result = dynamic_denoise_strength(
0.0, denoise_strength, [0, 0, 100, 100], (1000, 1000)
)
assert np.isclose(result, denoise_strength)
@given(