| """ |
| Tests for the Vision Module — ObjectDetector. |
| """ |
|
|
| import numpy as np |
| import pytest |
|
|
|
|
| class TestObjectDetector: |
| """Tests for ObjectDetector.detect() method.""" |
|
|
| def test_detect_returns_list(self): |
| """Detection should return a list.""" |
| from src.vision.detector import ObjectDetector |
|
|
| detector = ObjectDetector(model_name="yolov8n.pt", confidence=0.25) |
|
|
| |
| image = np.zeros((480, 640, 3), dtype=np.uint8) |
| image[:, :] = (200, 100, 50) |
|
|
| result = detector.detect(image) |
| assert isinstance(result, list), "detect() must return a list" |
|
|
| def test_detect_result_schema(self): |
| """Each detection should have label, confidence, and bbox keys.""" |
| from src.vision.detector import ObjectDetector |
|
|
| detector = ObjectDetector(model_name="yolov8n.pt", confidence=0.25) |
|
|
| |
| image = np.random.randint(0, 255, (480, 640, 3), dtype=np.uint8) |
| result = detector.detect(image) |
|
|
| for det in result: |
| assert "label" in det, "Detection must have 'label'" |
| assert "confidence" in det, "Detection must have 'confidence'" |
| assert "bbox" in det, "Detection must have 'bbox'" |
| assert isinstance(det["label"], str) |
| assert isinstance(det["confidence"], float) |
| assert isinstance(det["bbox"], list) |
| assert len(det["bbox"]) == 4 |
|
|
| def test_detect_invalid_image_raises(self): |
| """Passing None should raise ValueError.""" |
| from src.vision.detector import ObjectDetector |
|
|
| detector = ObjectDetector(model_name="yolov8n.pt", confidence=0.25) |
|
|
| with pytest.raises(ValueError): |
| detector.detect(None) |
|
|
| def test_detect_1d_image_raises(self): |
| """Passing a 1D array should raise ValueError.""" |
| from src.vision.detector import ObjectDetector |
|
|
| detector = ObjectDetector(model_name="yolov8n.pt", confidence=0.25) |
|
|
| with pytest.raises(ValueError): |
| detector.detect(np.array([1, 2, 3])) |
|
|
