Add tests for CLI

Makefile

@@ -12,9 +12,16 @@ format:
 	ruff format
 	ruff check --fix
 
+test:
+	python -m pytest -n auto -v tests
+
 clean:
 	rm -rf build dist *.egg-info
 
-publish: clean
+build: clean
 	python -m build --sdist --wheel
+
+upload: build
 	python -m twine upload dist/samuel-*
+
+publish: build upload

pyproject.toml

@@ -32,15 +32,18 @@ dependencies = [
   "fastapi",
   "loguru",
   "onnxruntime",
+  "Pillow",
   "uvicorn",
 ]
 dynamic = ["readme", "version"]
 
 [project.optional-dependencies]
 test = [
+  "build",
   "mypy",
   "types-Pillow",
   "pytest",
+  "pytest-xdist",
   "ruff",
   "twine",
 ]

tests/test___main__.py

@@ -0,0 +1,15 @@
+import io
+import os
+import shlex
+import subprocess
+
+import PIL.Image
+
+here = os.path.dirname(os.path.abspath(__file__))
+
+
+def test_run():
+    cmd = f"samuel run efficient-sam:10m --image {here}/../examples/_images/dogs.jpg"
+    output = subprocess.check_output(shlex.split(cmd))
+    image = PIL.Image.open(io.BytesIO(output))
+    assert image.size == (2560, 1600)