Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit e751243

Browse files
committedFeb 18, 2025·
fix: test cases
1 parent 2903135 commit e751243

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed
 

‎backend/tests/processing/test_process_queue.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,8 @@ def test_vectorize_extraction_process_step_single_reference(mock_chroma_db):
215215
# Assertions
216216
mock_vectorstore.add_docs.assert_called_once_with(
217217
docs=expected_docs,
218-
metadatas=expected_metadatas
218+
metadatas=expected_metadatas,
219+
batch_size=100
219220
)
220221

221222
@patch('app.processing.process_queue.ChromaDB')
@@ -261,7 +262,8 @@ def test_vectorize_extraction_process_step_multiple_references_concatenation(moc
261262
# Assertions
262263
mock_vectorstore.add_docs.assert_called_once_with(
263264
docs=expected_docs,
264-
metadatas=expected_metadatas
265+
metadatas=expected_metadatas,
266+
batch_size=100
265267
)
266268

267269
@patch('app.processing.process_queue.ChromaDB') # Replace with the correct module path

0 commit comments

Comments
 (0)
Please sign in to comment.