Skip to content

Commit dfecc3e

Browse files
committed
tests(chroma0): improve test coverages.
1 parent d311ac5 commit dfecc3e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/database/test_chroma0.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,8 @@ async def test_list_collection_content(mock_config):
145145
{
146146
"path": os.path.join(mock_config.project_root, "file1"),
147147
"sha256": "hash1",
148+
"start": 1,
149+
"end": 2,
148150
}
149151
],
150152
"documents": ["doc1"],
@@ -666,8 +668,9 @@ async def test_query_no_n_result(mock_config):
666668

667669

668670
@pytest.mark.asyncio
669-
async def test_create_or_get_collection_exists(mock_config):
671+
async def test_create_or_get_collection_exists(mock_config: Config):
670672
"""Test _create_or_get_collection when collection exists and allow_create is True."""
673+
mock_config.db_params["hnsw"] = {"M": 64}
671674
connector = ChromaDB0Connector(mock_config)
672675
with (
673676
patch(

0 commit comments

Comments
 (0)