mirror of
https://github.com/qdrant/qdrant-client.git
synced 2026-07-28 13:41:11 -05:00
97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
from tests.congruence_tests.test_common import (
|
|
COLLECTION_NAME,
|
|
compare_client_results,
|
|
compare_collections,
|
|
generate_fixtures,
|
|
generate_sparse_fixtures,
|
|
init_client,
|
|
init_local,
|
|
init_remote,
|
|
sparse_vectors_config,
|
|
)
|
|
|
|
|
|
def test_delete_points(local_client, remote_client):
|
|
points = generate_fixtures(100)
|
|
vector = points[0].vector["image"]
|
|
local_client.upload_points(COLLECTION_NAME, points)
|
|
remote_client.upload_points(COLLECTION_NAME, points, wait=True)
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="image").points,
|
|
)
|
|
|
|
found_ids = [
|
|
scored_point.id
|
|
for scored_point in local_client.query_points(
|
|
COLLECTION_NAME, query=vector, using="image"
|
|
).points
|
|
]
|
|
|
|
local_client.delete(COLLECTION_NAME, found_ids)
|
|
remote_client.delete(COLLECTION_NAME, found_ids)
|
|
|
|
compare_collections(local_client, remote_client, 100, attrs=("points_count",))
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="image").points,
|
|
)
|
|
|
|
# delete non-existent points
|
|
local_client.delete(COLLECTION_NAME, found_ids)
|
|
remote_client.delete(COLLECTION_NAME, found_ids)
|
|
|
|
compare_collections(local_client, remote_client, 100, attrs=("points_count",))
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="image").points,
|
|
)
|
|
|
|
|
|
def test_delete_sparse_points():
|
|
points = generate_sparse_fixtures(100)
|
|
vector = points[0].vector["sparse-image"]
|
|
|
|
local_client = init_local()
|
|
init_client(local_client, [], sparse_vectors_config=sparse_vectors_config)
|
|
|
|
remote_client = init_remote()
|
|
init_client(remote_client, [], sparse_vectors_config=sparse_vectors_config)
|
|
|
|
local_client.upload_points(COLLECTION_NAME, points)
|
|
remote_client.upload_points(COLLECTION_NAME, points, wait=True)
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.query_points(
|
|
COLLECTION_NAME,
|
|
query=vector,
|
|
using="sparse-image",
|
|
).points,
|
|
)
|
|
|
|
found_ids = [
|
|
scored_point.id
|
|
for scored_point in local_client.query_points(
|
|
COLLECTION_NAME, query=vector, using="sparse-image"
|
|
).points
|
|
]
|
|
|
|
local_client.delete(COLLECTION_NAME, found_ids)
|
|
remote_client.delete(COLLECTION_NAME, found_ids)
|
|
|
|
compare_collections(local_client, remote_client, 100, attrs=("points_count",))
|
|
|
|
compare_client_results(
|
|
local_client,
|
|
remote_client,
|
|
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="sparse-image").points,
|
|
)
|