Move image out of LFS for CI tests

- it's tiny, and the rest of the images aren't needed at the moment
Acly · Dec 5, 2023 · a7c38bd · a7c38bd
1 parent 2fc3f43
commit a7c38bd
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git 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
diff --git a/tests/images/outpaint_context.png b/tests/images/outpaint_context.png
diff --git 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: