feat: add CI timeout protection for tests

This commit is contained in:
Andy Lee
2025-08-07 23:54:56 -07:00
parent 777b5fed01
commit 0ec00e1a60
4 changed files with 137 additions and 1 deletions

View File

@@ -8,9 +8,11 @@ import tempfile
from pathlib import Path
import pytest
from test_timeout import ci_timeout
@pytest.mark.parametrize("backend_name", ["hnsw", "diskann"])
@ci_timeout(90) # 90 second timeout for this comprehensive test
def test_readme_basic_example(backend_name):
"""Test the basic example from README.md with both backends."""
# Skip on macOS CI due to MPS environment issues with all-MiniLM-L6-v2
@@ -79,6 +81,7 @@ def test_readme_imports():
assert callable(LeannChat)
@ci_timeout(60) # 60 second timeout
def test_backend_options():
"""Test different backend options mentioned in documentation."""
# Skip on macOS CI due to MPS environment issues with all-MiniLM-L6-v2
@@ -115,6 +118,7 @@ def test_backend_options():
@pytest.mark.parametrize("backend_name", ["hnsw", "diskann"])
@ci_timeout(75) # 75 second timeout for LLM tests
def test_llm_config_simulated(backend_name):
"""Test simulated LLM configuration option with both backends."""
# Skip on macOS CI due to MPS environment issues with all-MiniLM-L6-v2