Skip to content

Commit bf8ff0a

Browse files
committed
fix test warnings
Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>
1 parent 58207be commit bf8ff0a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_vector_db_weaviate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ async def test_write_documents_default_embedding(self) -> None:
269269
):
270270
mock_client = AsyncMock()
271271
mock_collection = AsyncMock()
272-
mock_batch = AsyncMock()
272+
mock_batch = MagicMock()
273273
mock_batch_context = AsyncMock()
274274

275275
mock_client.collections.exists = AsyncMock(return_value=True)
@@ -318,7 +318,7 @@ async def test_write_documents_custom_embedding(self) -> None:
318318
):
319319
mock_client = AsyncMock()
320320
mock_collection = AsyncMock()
321-
mock_batch = AsyncMock()
321+
mock_batch = MagicMock()
322322
mock_batch_context = AsyncMock()
323323

324324
mock_client.collections.exists = AsyncMock(return_value=False)
@@ -370,7 +370,7 @@ async def test_write_documents_ignores_per_write_embedding_with_warning(
370370
):
371371
mock_client = AsyncMock()
372372
mock_collection = AsyncMock()
373-
mock_batch = AsyncMock()
373+
mock_batch = MagicMock()
374374
mock_batch_context = AsyncMock()
375375

376376
mock_client.collections.exists = AsyncMock(return_value=True)

0 commit comments

Comments
 (0)