Add lightweight test mode tests for IntelligentContentAnalyzer and ModelRegistry

Author: smirk-dev

.github/workflows/tests.yml

@@ -16,14 +16,21 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.11'
+      - name: Cache pip
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/pip
+          key: ${{ runner.os }}-pip-fast-${{ hashFiles('requirements-test.txt') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-fast-
       - name: Install minimal deps
         run: |
           python -m pip install --upgrade pip
           pip install -r requirements-test.txt
-      - name: Run lightweight tests
+      - name: Run lightweight tests with coverage
         env:
           LIGHT_TEST_MODE: '1'
-        run: pytest -q
+        run: pytest -q --cov=src --cov-report=term-missing:skip-covered --maxfail=1 -m "not heavy"
 
   full-tests:
     if: github.event_name == 'schedule'
@@ -33,11 +40,18 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.11'
+      - name: Cache pip
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/pip
+          key: ${{ runner.os }}-pip-full-${{ hashFiles('requirements.txt') }}
+          restore-keys: |
+            ${{ runner.os }}-pip-full-
       - name: Install full deps
         run: |
             python -m pip install --upgrade pip
             pip install -r requirements.txt
       - name: Run integration tests
         env:
           LIGHT_TEST_MODE: '0'
-        run: pytest -q || true  # tolerate failures initially
+        run: pytest -q -m "integration or heavy" || true  # tolerate failures initially

requirements-test.txt

@@ -1,6 +1,7 @@
 # Minimal dependencies for running lightweight tests
 pytest>=8.0.0
 pytest-asyncio>=0.21.0
+pytest-cov>=5.0.0
 PyYAML>=6.0.0
 streamlit>=1.32.0
 opencv-python-headless>=4.8.0

tests/conftest.py

@@ -6,9 +6,25 @@
 """
 import os
 import pytest
+import pytest
 
 def pytest_configure():  # pragma: no cover
     os.environ.setdefault('LIGHT_TEST_MODE', '1')
+    # Register custom markers to avoid warnings on CI output
+    for marker in [
+        'heavy: marks tests that load large ML models (deselected in fast CI)',
+        'integration: multi-component integration tests',
+        'slow: longer-running tests',
+    ]:
+        try:  # pytest <8 compatibility
+            pytest.config.addinivalue_line('markers', marker)  # type: ignore[attr-defined]
+        except Exception:  # pragma: no cover
+            pass
+
+
+def pytest_runtest_setup(item):  # pragma: no cover
+    if os.getenv('LIGHT_TEST_MODE') == '1' and 'heavy' in item.keywords:
+        pytest.skip('Skipping heavy test in LIGHT_TEST_MODE=1')
     # Register custom markers to avoid warnings
     for marker in [
         'heavy: marks tests that load large ML models (deselected in fast CI)',

tests/test_intelligent_content_analyzer_light.py

@@ -0,0 +1,19 @@
+from src.ai_models.intelligent_content_analyzer import IntelligentContentAnalyzer
+
+
+def test_content_analyzer_light_mode(monkeypatch):
+    monkeypatch.setenv('LIGHT_TEST_MODE', '1')
+    analyzer = IntelligentContentAnalyzer({'ai_models': {'cache_dir': './cache'}})
+    assert analyzer.content_classifier is None
+    assert analyzer.object_detector is None
+    assert analyzer.text_analyzer is None
+
+
+def test_content_analyzer_basic_analysis(monkeypatch, tmp_path):
+    monkeypatch.setenv('LIGHT_TEST_MODE', '1')
+    analyzer = IntelligentContentAnalyzer({'ai_models': {'cache_dir': './cache'}})
+    dummy_video = tmp_path / 'dummy.mp4'
+    dummy_video.write_bytes(b'')
+    result = analyzer.analyze_content(str(dummy_video))
+    assert 'content_type' in result
+    assert 'editing_recommendations' in result

tests/test_model_registry.py

@@ -0,0 +1,16 @@
+import pytest
+
+from src.ai_models import ModelRegistry
+
+
+def test_registry_light_mode(monkeypatch):
+    monkeypatch.setenv('LIGHT_TEST_MODE', '1')
+    pipe = ModelRegistry.get_pipeline('text-classification', 'distilbert-base-uncased')
+    assert pipe is None
+
+
+@pytest.mark.heavy
+def test_registry_attempt_load(monkeypatch):
+    monkeypatch.setenv('LIGHT_TEST_MODE', '0')
+    pipe = ModelRegistry.get_pipeline('text-classification', 'distilbert-base-uncased')
+    assert pipe is None or hasattr(pipe, '__call__')