chore: 允许 ai-service Python 文件入库(.gitignore 例外)

This commit is contained in:
weijuesen
2026-08-12 16:22:45 +08:00
parent 99137a6fb4
commit b89ad5e865
9 changed files with 278 additions and 0 deletions
+25
View File
@@ -0,0 +1,25 @@
import base64
import pytest
from app.detector import MockDetector
# 1x1 透明 PNG
TINY_PNG = base64.b64decode(
"iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg=="
)
def test_mock_detector_returns_canned_detection():
det = MockDetector(class_name="sick", confidence=0.92)
result = det.detect(TINY_PNG)
assert len(result) == 1
assert result[0]["class_name"] == "sick"
assert result[0]["confidence"] == 0.92
assert result[0]["bbox"]["w"] > 0
def test_mock_detector_rejects_invalid_image():
det = MockDetector()
with pytest.raises(ValueError):
det.detect(b"not an image")