diff --git a/tests/config.py b/tests/config.py index fe823c0150..050362e899 100644 --- a/tests/config.py +++ b/tests/config.py @@ -4,6 +4,7 @@ test_dir = Path(__file__).parent server_dir = test_dir / ".server" +data_dir = test_dir / "data" image_dir = test_dir / "images" result_dir = test_dir / ".results" reference_dir = test_dir / "references" diff --git a/tests/data/outpaint_context.png b/tests/data/outpaint_context.png new file mode 100644 index 0000000000..2b9aa17b4f Binary files /dev/null and b/tests/data/outpaint_context.png differ diff --git a/tests/images/outpaint_context.png b/tests/images/outpaint_context.png deleted file mode 100644 index d76b687b6b..0000000000 --- a/tests/images/outpaint_context.png +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:899c7078792c4d10ce22b9a22f2a220790d504fd4256d764a3595acc77a16133 -size 1734 diff --git a/tests/test_workflow.py b/tests/test_workflow.py index afb401817d..784415f771 100644 --- a/tests/test_workflow.py +++ b/tests/test_workflow.py @@ -8,7 +8,7 @@ from ai_diffusion.pose import Pose from ai_diffusion.workflow import LiveParams, Conditioning, Control from pathlib import Path -from .config import image_dir, result_dir, reference_dir, default_checkpoint +from .config import data_dir, image_dir, result_dir, reference_dir, default_checkpoint @pytest.fixture(scope="session", autouse=True) @@ -86,7 +86,7 @@ def test_compute_batch_size(extent, min_size, max_batches, expected): ids=["left", "right", "top", "bottom", "full", "small", "offset"], ) def test_inpaint_context(area, expected_extent, expected_crop: tuple[int, int] | None): - image = Image.load(image_dir / "outpaint_context.png") + image = Image.load(data_dir / "outpaint_context.png") default = comfyworkflow.Output(0, 0) result = workflow.create_inpaint_context(image, area, default) if expected_crop: