connor-jason
commited on
Commit
·
b634622
1
Parent(s):
548a2bb
is this a pytest
Browse files- .github/workflows/run-tests.yml +1 -2
- pyproject.toml +5 -0
- tests/__init__.py +0 -0
- tests/test_models.py +19 -0
.github/workflows/run-tests.yml
CHANGED
|
@@ -13,10 +13,9 @@ jobs:
|
|
| 13 |
|
| 14 |
steps:
|
| 15 |
- uses: actions/checkout@v6
|
| 16 |
-
|
| 17 |
|
| 18 |
- name: Install uv
|
| 19 |
-
|
| 20 |
|
| 21 |
- name: Install Project
|
| 22 |
run: uv sync --locked --all-extras --dev
|
|
|
|
| 13 |
|
| 14 |
steps:
|
| 15 |
- uses: actions/checkout@v6
|
|
|
|
| 16 |
|
| 17 |
- name: Install uv
|
| 18 |
+
uses: astral-sh/setup-uv@v7
|
| 19 |
|
| 20 |
- name: Install Project
|
| 21 |
run: uv sync --locked --all-extras --dev
|
pyproject.toml
CHANGED
|
@@ -15,3 +15,8 @@ dependencies = [
|
|
| 15 |
"unsloth>=2025.9.9",
|
| 16 |
"qwen-vl-utils>=0.0.14",
|
| 17 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
"unsloth>=2025.9.9",
|
| 16 |
"qwen-vl-utils>=0.0.14",
|
| 17 |
]
|
| 18 |
+
|
| 19 |
+
[dependency-groups]
|
| 20 |
+
dev = [
|
| 21 |
+
"pytest>=8.3.4",
|
| 22 |
+
]
|
tests/__init__.py
ADDED
|
File without changes
|
tests/test_models.py
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from PIL import Image
|
| 2 |
+
from local_model import query_local
|
| 3 |
+
from remote_model import query_remote, client
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
def test_local_model():
|
| 7 |
+
img = Image.new('RGB', (100, 100), color='red')
|
| 8 |
+
result = query_local(img, "test!")
|
| 9 |
+
|
| 10 |
+
assert isinstance(result, str)
|
| 11 |
+
assert len(result) > 0
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
def test_remote_model():
|
| 15 |
+
img = Image.new('RGB', (100, 100), color='blue')
|
| 16 |
+
result = query_remote(img, "test?", client)
|
| 17 |
+
|
| 18 |
+
assert isinstance(result, str)
|
| 19 |
+
assert len(result) > 0
|