test coverage

Author: Zhe Yu

src/vectorcode/mcp_main.py

@@ -161,7 +161,7 @@ async def vectorise_files(paths: list[str], project_root: str) -> dict[str, int]
             await remove_orphanes(collection, collection_lock, stats, stats_lock)
 
         return stats.to_dict()
-    except Exception as e:
+    except Exception as e:  # pragma: nocover
         if isinstance(e, McpError):
             logger.error("Failed to access collection at %s", project_root)
             raise
@@ -225,7 +225,7 @@ async def query_tool(
             logger.info("Retrieved the following files: %s", result_paths)
             return results
 
-    except Exception as e:
+    except Exception as e:  # pragma: nocover
         if isinstance(e, McpError):
             logger.error("Failed to access collection at %s", project_root)
             raise

tests/subcommands/query/test_reranker.py

@@ -4,7 +4,7 @@
 import numpy
 import pytest
 
-from vectorcode.cli_utils import Config
+from vectorcode.cli_utils import Config, QueryInclude
 from vectorcode.subcommands.query.reranker import (
     CrossEncoderReranker,
     NaiveReranker,
@@ -129,6 +129,22 @@ async def test_naive_reranker_rerank(naive_reranker_conf, query_result):
         assert isinstance(res, str)
 
 
+@pytest.mark.asyncio
+async def test_naive_reranker_rerank_chunks(naive_reranker_conf, query_result):
+    """Test basic reranking functionality of NaiveReranker"""
+    naive_reranker_conf.include = [QueryInclude.chunk]
+    reranker = NaiveReranker(naive_reranker_conf)
+    chunk_text = {str(i.chunk) for i in query_result}
+    result = await reranker.rerank(query_result)
+
+    # Check the result is a list of paths with correct length
+    assert isinstance(result, list)
+    assert len(result) <= naive_reranker_conf.n_result
+
+    for res in result:
+        assert res in chunk_text
+
+
 @pytest.mark.asyncio
 async def test_naive_reranker_rerank_empty_result(
     naive_reranker_conf, empty_query_result