mirror of
https://github.com/qdrant/qdrant-client.git
synced 2026-07-30 22:51:03 -05:00
392e3b6d1f
* base class for qdrant * WIP: implement local qdrant client * fix mypy * fix mypy * search tests * search tests * fix mypy * scroll test * filters: fixtures, tests, and fixes * fix types * fix types * fix types * fix: fix local CollectionInfo, add __test__ to avoid pytest complaints, fix typo * fix: fix typo in import * tests: add local upload tests (#139) * new: make local collection info more similar to remote one * new: add local upsert tests * tests: moved compare collections * scroll tests * recommendations test * tests: fix vector comparison in utils, add retrieve tests * fix: fix types * persistence test * fix: fix types * fix: fix db path creation * tests: add delete points tests, refactoring (#140) * skip local tests on old version * test aliases * test count * tests: add delete payload tests, move set and overwrite payload tests * tests: fix pytest warning * cover some more stuff with tests --------- Co-authored-by: George Panchuk <george.panchuk@qdrant.tech>
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from qdrant_client.http.models import NamedVector
|
|
from tests.congruence_tests.test_common import (
|
|
COLLECTION_NAME,
|
|
compare_client_results,
|
|
compare_collections,
|
|
generate_fixtures,
|
|
)
|
|
|
|
|
|
def test_delete_points(local_client, remote_client):
|
|
records = generate_fixtures(100)
|
|
vector = records[0].vector["image"]
|
|
|
|
local_client.upload_records(COLLECTION_NAME, records)
|
|
remote_client.upload_records(COLLECTION_NAME, records)
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.search(COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)),
|
|
)
|
|
|
|
found_ids = [
|
|
scored_point.id
|
|
for scored_point in local_client.search(
|
|
COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)
|
|
)
|
|
]
|
|
|
|
local_client.delete(COLLECTION_NAME, found_ids)
|
|
remote_client.delete(COLLECTION_NAME, found_ids)
|
|
|
|
compare_collections(local_client, remote_client, 100)
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.search(COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)),
|
|
)
|