Skip to content

Commit 4e3f8da

Browse files
committed
fixing tests for async version
1 parent 5282c20 commit 4e3f8da

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

integrations/qdrant/tests/test_document_store_async.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -274,10 +274,9 @@ async def test_delete_by_filter_async(self, document_store: QdrantDocumentStore)
274274
assert await document_store.count_documents_async() == 3
275275

276276
# Delete documents with category="A"
277-
deleted_count = await document_store.delete_by_filter_async(
277+
await document_store.delete_by_filter_async(
278278
filters={"field": "meta.category", "operator": "==", "value": "A"}
279279
)
280-
assert deleted_count == 2
281280
assert await document_store.count_documents_async() == 1
282281

283282
# Verify only category B remains
@@ -288,10 +287,9 @@ async def test_delete_by_filter_async(self, document_store: QdrantDocumentStore)
288287
assert remaining_docs[0].meta["category"] == "B"
289288

290289
# Delete remaining document by year
291-
deleted_count = await document_store.delete_by_filter_async(
290+
await document_store.delete_by_filter_async(
292291
filters={"field": "meta.year", "operator": "==", "value": 2023}
293292
)
294-
assert deleted_count == 1
295293
assert await document_store.count_documents_async() == 0
296294

297295
@pytest.mark.asyncio
@@ -304,10 +302,9 @@ async def test_delete_by_filter_async_no_matches(self, document_store: QdrantDoc
304302
assert await document_store.count_documents_async() == 2
305303

306304
# Try to delete documents with category="C" (no matches)
307-
deleted_count = await document_store.delete_by_filter_async(
305+
await document_store.delete_by_filter_async(
308306
filters={"field": "meta.category", "operator": "==", "value": "C"}
309307
)
310-
assert deleted_count == 0
311308
assert await document_store.count_documents_async() == 2
312309

313310
@pytest.mark.asyncio
@@ -321,7 +318,7 @@ async def test_delete_by_filter_async_advanced_filters(self, document_store: Qdr
321318
assert await document_store.count_documents_async() == 3
322319

323320
# AND condition
324-
deleted_count = await document_store.delete_by_filter_async(
321+
await document_store.delete_by_filter_async(
325322
filters={
326323
"operator": "AND",
327324
"conditions": [
@@ -330,11 +327,10 @@ async def test_delete_by_filter_async_advanced_filters(self, document_store: Qdr
330327
],
331328
}
332329
)
333-
assert deleted_count == 1
334330
assert await document_store.count_documents_async() == 2
335331

336332
# OR condition
337-
deleted_count = await document_store.delete_by_filter_async(
333+
await document_store.delete_by_filter_async(
338334
filters={
339335
"operator": "OR",
340336
"conditions": [
@@ -343,7 +339,6 @@ async def test_delete_by_filter_async_advanced_filters(self, document_store: Qdr
343339
],
344340
}
345341
)
346-
assert deleted_count == 2
347342
assert await document_store.count_documents_async() == 0
348343

349344
@pytest.mark.asyncio

0 commit comments

Comments
 (0)