mirror of
https://github.com/ruvnet/RuView.git
synced 2026-04-28 05:59:32 +00:00
5bd0d59aa6
P1 fixes (this sprint): - P1-6: Extract sensing-server modules (cli, types, csi, pose) from main.rs - P1-7: DDA ray march for tomography — O(max(n)) replaces O(n^3) voxel scan - P1-8: Batch neural inference — Tensor::stack/split for single GPU call - P1-10: Eliminate 112KB/frame alloc — islice replaces deque→list copy P2 fixes (this quarter): - P2-11: Python unit tests for 8 modules (rate_limit, auth, error_handler, pose_service, stream_service, hardware_service, health_check, metrics) - P2-13: MAT simulated data safety guard — blocking overlay + pulsing banner - P2-14: Wire token blacklist into auth verification + logout endpoint - P2-15: Frame budget benchmark — confirms pipeline well under 50ms budget Addresses 8 of 10 remaining issues from QE analysis (ADR-080). Co-Authored-By: claude-flow <ruv@ruv.net>
73 lines
2.9 KiB
Python
73 lines
2.9 KiB
Python
"""Tests for PoseService."""
|
|
|
|
import pytest
|
|
import asyncio
|
|
from unittest.mock import MagicMock, AsyncMock, patch
|
|
from datetime import datetime
|
|
|
|
|
|
class TestPoseServiceInit:
|
|
def test_init_sets_defaults(self, mock_settings, mock_domain_config):
|
|
with patch.dict("sys.modules", {
|
|
"torch": MagicMock(),
|
|
"src.models.densepose_head": MagicMock(),
|
|
"src.models.modality_translation": MagicMock(),
|
|
}):
|
|
from src.services.pose_service import PoseService
|
|
svc = PoseService(mock_settings, mock_domain_config)
|
|
assert svc.is_initialized is False
|
|
assert svc.is_running is False
|
|
assert svc.stats["total_processed"] == 0
|
|
|
|
def test_stats_are_zero_on_init(self, mock_settings, mock_domain_config):
|
|
with patch.dict("sys.modules", {
|
|
"torch": MagicMock(),
|
|
"src.models.densepose_head": MagicMock(),
|
|
"src.models.modality_translation": MagicMock(),
|
|
}):
|
|
from src.services.pose_service import PoseService
|
|
svc = PoseService(mock_settings, mock_domain_config)
|
|
assert svc.stats["successful_detections"] == 0
|
|
assert svc.stats["failed_detections"] == 0
|
|
assert svc.stats["average_confidence"] == 0.0
|
|
|
|
|
|
class TestPoseServiceLifecycle:
|
|
@pytest.mark.asyncio
|
|
async def test_initialize_sets_flag(self, mock_settings, mock_domain_config):
|
|
with patch.dict("sys.modules", {
|
|
"torch": MagicMock(),
|
|
"src.models.densepose_head": MagicMock(),
|
|
"src.models.modality_translation": MagicMock(),
|
|
}):
|
|
from src.services.pose_service import PoseService
|
|
svc = PoseService(mock_settings, mock_domain_config)
|
|
await svc.initialize()
|
|
assert svc.is_initialized is True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_start_stop(self, mock_settings, mock_domain_config):
|
|
with patch.dict("sys.modules", {
|
|
"torch": MagicMock(),
|
|
"src.models.densepose_head": MagicMock(),
|
|
"src.models.modality_translation": MagicMock(),
|
|
}):
|
|
from src.services.pose_service import PoseService
|
|
svc = PoseService(mock_settings, mock_domain_config)
|
|
await svc.initialize()
|
|
await svc.start()
|
|
assert svc.is_running is True
|
|
await svc.stop()
|
|
assert svc.is_running is False
|
|
|
|
|
|
class TestPoseServiceStats:
|
|
def test_initial_classification(self, mock_settings, mock_domain_config):
|
|
with patch.dict("sys.modules", {
|
|
"torch": MagicMock(),
|
|
"src.models.densepose_head": MagicMock(),
|
|
"src.models.modality_translation": MagicMock(),
|
|
}):
|
|
from src.services.pose_service import PoseService
|
|
svc = PoseService(mock_settings, mock_domain_config)
|
|
assert svc.last_error is None
|