diff --git a/apps/coordinator/tests/test_context_compaction.py b/apps/coordinator/tests/test_context_compaction.py index 0573eb6..af7c761 100644 --- a/apps/coordinator/tests/test_context_compaction.py +++ b/apps/coordinator/tests/test_context_compaction.py @@ -6,11 +6,11 @@ Context compaction reduces memory usage by: 3. Measuring context reduction achieved """ -import pytest -from unittest.mock import AsyncMock, MagicMock +from unittest.mock import AsyncMock -from src.context_compaction import ContextCompactor, CompactionResult -from src.models import ContextUsage +import pytest + +from src.context_compaction import CompactionResult, ContextCompactor class TestContextCompactor: diff --git a/apps/coordinator/tests/test_context_monitor.py b/apps/coordinator/tests/test_context_monitor.py index e3e2f4d..cacfe6f 100644 --- a/apps/coordinator/tests/test_context_monitor.py +++ b/apps/coordinator/tests/test_context_monitor.py @@ -325,6 +325,7 @@ class TestContextMonitor: ) -> None: """Should perform compaction when COMPACT action is triggered.""" from unittest.mock import patch + from src.context_compaction import CompactionResult # Mock compaction result @@ -365,6 +366,7 @@ class TestContextMonitor: ) -> None: """Should log compaction metrics when compaction is performed.""" from unittest.mock import patch + from src.context_compaction import CompactionResult mock_compaction_result = CompactionResult( @@ -402,6 +404,7 @@ class TestContextMonitor: ) -> None: """Should handle compaction failure and log error.""" from unittest.mock import patch + from src.context_compaction import CompactionResult mock_compaction_result = CompactionResult(