fix: use session level loop for e2e tests

Author: georgeh0

tests/conftest.py

@@ -2,27 +2,32 @@
 
 import os
 import tempfile
+from collections.abc import AsyncIterator
 from pathlib import Path
 
+import cocoindex.asyncio as coco_aio
 import pytest
+import pytest_asyncio
 
-# === Environment setup BEFORE any imports ===
-# This must happen before cocoindex_code modules are imported
-
-
-# Use tiny model for faster tests
-# os.environ["COCOINDEX_CODE_EMBEDDING_MODEL"] = "sentence-transformers/paraphrase-MiniLM-L3-v2"
-
+# === Environment setup BEFORE any cocoindex_code imports ===
 # Create test directory and set it BEFORE any module imports
 _TEST_DIR = Path(tempfile.mkdtemp(prefix="cocoindex_test_"))
 os.environ["COCOINDEX_CODE_ROOT_PATH"] = str(_TEST_DIR)
-os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
-
-# Set PyTorch to single thread to avoid potential failure on testing environment
-# torch.set_num_threads(1)
 
 
 @pytest.fixture(scope="session")
 def test_codebase_root() -> Path:
     """Session-scoped test codebase directory."""
     return _TEST_DIR
+
+
+@pytest_asyncio.fixture(scope="session", loop_scope="session")
+async def coco_runtime() -> AsyncIterator[None]:
+    """
+    Set up CocoIndex runtime context for the entire test session.
+
+    Uses session-scoped event loop to ensure CocoIndex environment
+    persists across all tests.
+    """
+    async with coco_aio.runtime():
+        yield

tests/test_e2e.py

@@ -2,9 +2,7 @@
 
 import shutil
 from pathlib import Path
-from typing import  Iterator
 
-import cocoindex as coco
 import pytest
 
 from cocoindex_code.config import _discover_codebase_root
@@ -13,13 +11,6 @@
 
 pytest_plugins = ("pytest_asyncio",)
 
-
-@pytest.fixture(scope="module")
-def coco_runtime() -> Iterator[None]:
-    """Set up CocoIndex runtime context shared across all tests in this module."""
-    with coco.runtime():
-        yield
-
 # === Sample codebase files ===
 
 SAMPLE_MAIN_PY = '''\
@@ -140,7 +131,7 @@ def setup_base_codebase(codebase: Path) -> None:
 class TestEndToEnd:
     """End-to-end tests for the complete index-query workflow."""
 
-    @pytest.mark.asyncio
+    @pytest.mark.asyncio(loop_scope="session")
     async def test_index_and_query_codebase(
         self, test_codebase_root: Path, coco_runtime: None
     ) -> None:
@@ -164,7 +155,7 @@ async def test_index_and_query_codebase(
         assert len(results) > 0
         assert "database.py" in results[0].file_path
 
-    @pytest.mark.asyncio
+    @pytest.mark.asyncio(loop_scope="session")
     async def test_incremental_update_add_file(
         self, test_codebase_root: Path, coco_runtime: None
     ) -> None:
@@ -190,7 +181,7 @@ async def test_incremental_update_add_file(
         assert len(results) > 0
         assert "ml_model.py" in results[0].file_path
 
-    @pytest.mark.asyncio
+    @pytest.mark.asyncio(loop_scope="session")
     async def test_incremental_update_modify_file(
         self, test_codebase_root: Path, coco_runtime: None
     ) -> None:
@@ -210,7 +201,7 @@ async def test_incremental_update_modify_file(
         content_lower = results[0].content.lower()
         assert "authenticate" in content_lower or "login" in content_lower
 
-    @pytest.mark.asyncio
+    @pytest.mark.asyncio(loop_scope="session")
     async def test_incremental_update_delete_file(
         self, test_codebase_root: Path, coco_runtime: None
     ) -> None: