Skip to content

Commit af10ff9

Browse files
committed
updating tests
1 parent 5dc1bac commit af10ff9

1 file changed

Lines changed: 4 additions & 14 deletions

File tree

integrations/qdrant/tests/test_document_store.py

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -348,13 +348,6 @@ def test_delete_by_filter(self, document_store: QdrantDocumentStore):
348348
document_store.write_documents(docs)
349349
assert document_store.count_documents() == 3
350350

351-
# Delete documents with category="A"
352-
deleted_count = document_store.delete_by_filter(
353-
filters={"field": "meta.category", "operator": "==", "value": "A"}
354-
)
355-
assert deleted_count == 2
356-
assert document_store.count_documents() == 1
357-
358351
# Verify only category B remains
359352
remaining_docs = document_store.filter_documents()
360353
assert len(remaining_docs) == 1
@@ -373,11 +366,10 @@ def test_delete_by_filter_no_matches(self, document_store: QdrantDocumentStore):
373366
document_store.write_documents(docs)
374367
assert document_store.count_documents() == 2
375368

376-
# Try to delete documents with category="C" (no matches)
377-
deleted_count = document_store.delete_by_filter(
369+
# try to delete documents with category="C" (no matches)
370+
document_store.delete_by_filter(
378371
filters={"field": "meta.category", "operator": "==", "value": "C"}
379372
)
380-
assert deleted_count == 0
381373
assert document_store.count_documents() == 2
382374

383375
def test_delete_by_filter_advanced_filters(self, document_store: QdrantDocumentStore):
@@ -390,7 +382,7 @@ def test_delete_by_filter_advanced_filters(self, document_store: QdrantDocumentS
390382
assert document_store.count_documents() == 3
391383

392384
# AND condition
393-
deleted_count = document_store.delete_by_filter(
385+
document_store.delete_by_filter(
394386
filters={
395387
"operator": "AND",
396388
"conditions": [
@@ -399,11 +391,10 @@ def test_delete_by_filter_advanced_filters(self, document_store: QdrantDocumentS
399391
],
400392
}
401393
)
402-
assert deleted_count == 1
403394
assert document_store.count_documents() == 2
404395

405396
# OR condition
406-
deleted_count = document_store.delete_by_filter(
397+
document_store.delete_by_filter(
407398
filters={
408399
"operator": "OR",
409400
"conditions": [
@@ -412,7 +403,6 @@ def test_delete_by_filter_advanced_filters(self, document_store: QdrantDocumentS
412403
],
413404
}
414405
)
415-
assert deleted_count == 2
416406
assert document_store.count_documents() == 0
417407

418408
def test_update_by_filter(self, document_store: QdrantDocumentStore):

0 commit comments

Comments
 (0)