Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ruff Formatting

Browse Source 
Fix pyTest issues
This commit is contained in:
skunkworxdark 2023-12-08 18:31:33 +00:00
parent fdb97c1d02
commit d3ad356c6a
2 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/backend/tiles/tiles.py
View File

@ -159,7 +159,12 @@ def calc_tiles_even_split(
def calc_tiles_min_overlap(
image_height: int, image_width: int, tile_height: int, tile_width: int, min_overlap: int = 0, round_to_8: bool = False
image_height: int,
image_width: int,
tile_height: int,
tile_width: int,
min_overlap: int = 0,
round_to_8: bool = False,
) -> list[Tile]:
"""Calculate the tile coordinates for a given image shape under a simple tiling scheme with overlaps.

13
tests/backend/tiles/test_tiles.py
View File

@ -223,7 +223,7 @@ def test_calc_tiles_min_overlap_difficult_size_div8():
@pytest.mark.parametrize(
["image_height", "image_width", "tile_height", "tile_width", "overlap", "raises"],
["image_height", "image_width", "tile_height", "tile_width", "min_overlap", "raises"],
[
(128, 128, 128, 128, 127, False), # OK
(128, 128, 128, 128, 0, False), # OK
@ -240,15 +240,14 @@ def test_calc_tiles_min_overlap_input_validation(
tile_height: int,
tile_width: int,
min_overlap: int,
round_to_8: bool,
raises: bool,
):
"""Test that calc_tiles_with_overlap() raises an exception if the inputs are invalid."""
"""Test that calc_tiles_min_overlap() raises an exception if the inputs are invalid."""
if raises:
with pytest.raises(AssertionError):
calc_tiles_min_overlap(image_height, image_width, tile_height, tile_width, min_overlap, round_to_8)
calc_tiles_min_overlap(image_height, image_width, tile_height, tile_width, min_overlap)
else:
calc_tiles_min_overlap(image_height, image_width, tile_height, tile_width, min_overlap, round_to_8)
calc_tiles_min_overlap(image_height, image_width, tile_height, tile_width, min_overlap)
####################################
@ -333,13 +332,13 @@ def test_calc_tiles_even_split_difficult_size():
(128, 128, 1, 1, 0.25, False), # OK
(128, 128, 1, 1, 0, False), # OK
(128, 128, 2, 1, 0, False), # OK
(127, 127, 1, 1, 0, True), # image size must be drivable by 8
(127, 127, 1, 1, 0, True), # image size must be dividable by 8
],
)
def test_calc_tiles_even_split_input_validation(
image_height: int, image_width: int, num_tiles_x: int, num_tiles_y: int, overlap: float, raises: bool
):
"""Test that calc_tiles_with_overlap() raises an exception if the inputs are invalid."""
"""Test that calc_tiles_even_split() raises an exception if the inputs are invalid."""
if raises:
with pytest.raises(AssertionError):
calc_tiles_even_split(image_height, image_width, num_tiles_x, num_tiles_y, overlap)