fix: explicitly shutdown in pytest fixture

Author: georgeh0

python/cocoindex/subprocess_exec.py

@@ -37,7 +37,7 @@
 _logger = logging.getLogger(__name__)
 
 
-def _shutdown_pool_at_exit() -> None:
+def shutdown_pool_at_exit() -> None:
     """Best-effort shutdown of the global ProcessPoolExecutor on interpreter exit."""
     global _pool, _pool_cleanup_registered  # pylint: disable=global-statement
     print("Shutting down pool at exit")
@@ -65,7 +65,7 @@ def _get_pool() -> ProcessPoolExecutor:
             if not _pool_cleanup_registered:
                 # Register the shutdown at exit at creation time (rather than at import time)
                 # to make sure it's executed earlier in the shutdown sequence.
-                atexit.register(_shutdown_pool_at_exit)
+                atexit.register(shutdown_pool_at_exit)
                 _pool_cleanup_registered = True
 
             # Single worker process as requested

python/cocoindex/tests/conftest.py

@@ -0,0 +1,15 @@
+import pytest
+import typing
+
+
+@pytest.fixture(scope="session", autouse=True)
+def _cocoindex_env_fixture() -> typing.Generator[None, None, None]:
+    """Shutdown the subprocess pool at exit."""
+    yield
+    try:
+        print("Shutdown the subprocess pool at exit in hook.")
+        import cocoindex.subprocess_exec
+
+        cocoindex.subprocess_exec.shutdown_pool_at_exit()
+    except Exception:
+        pass