chore: 允许 ai-service Python 文件入库(.gitignore 例外)
This commit is contained in:
@@ -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")
|
||||
Reference in New Issue
Block a user