mirror of
https://github.com/qdrant/qdrant-client.git
synced 2026-07-29 22:21:10 -05:00
a2adc2d2ad
* new: deprecate upload records, update tests, prohibit migration of collections with custom shards * Update qdrant_client/qdrant_client.py * Update qdrant_client/async_qdrant_client.py * new: replace autogenerated int ids with uuids (#448) * fix: remove redundant kwargs * fix: regen async * fix: remove redundant import --------- Co-authored-by: Andrey Vasnetsov <andrey@vasnetsov.com>
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
from qdrant_client.client_base import QdrantBase
|
|
from qdrant_client.http import models
|
|
from tests.congruence_tests.test_common import (
|
|
COLLECTION_NAME,
|
|
compare_client_results,
|
|
generate_fixtures,
|
|
generate_sparse_fixtures,
|
|
init_client,
|
|
init_local,
|
|
init_remote,
|
|
sparse_vectors_config,
|
|
)
|
|
from tests.fixtures.filters import one_random_filter_please
|
|
|
|
|
|
def count_all(client: QdrantBase) -> int:
|
|
return client.count(
|
|
collection_name=COLLECTION_NAME,
|
|
count_filter=None,
|
|
).count
|
|
|
|
|
|
def filter_count(client: QdrantBase, count_filter: models.Filter) -> int:
|
|
return client.count(
|
|
collection_name=COLLECTION_NAME,
|
|
count_filter=count_filter,
|
|
).count
|
|
|
|
|
|
def test_simple_count():
|
|
fixture_points = generate_fixtures()
|
|
|
|
local_client = init_local()
|
|
init_client(local_client, fixture_points)
|
|
|
|
remote_client = init_remote()
|
|
init_client(remote_client, fixture_points)
|
|
|
|
compare_client_results(local_client, remote_client, count_all)
|
|
|
|
for i in range(100):
|
|
count_filter = one_random_filter_please()
|
|
try:
|
|
compare_client_results(
|
|
local_client, remote_client, filter_count, count_filter=count_filter
|
|
)
|
|
except AssertionError as e:
|
|
print(f"\nFailed with filter {count_filter}")
|
|
raise e
|
|
|
|
|
|
def test_simple_sparse_search():
|
|
fixture_points = generate_sparse_fixtures()
|
|
|
|
local_client = init_local()
|
|
init_client(local_client, fixture_points, sparse_vectors_config=sparse_vectors_config)
|
|
|
|
remote_client = init_remote()
|
|
init_client(remote_client, fixture_points, sparse_vectors_config=sparse_vectors_config)
|
|
|
|
compare_client_results(local_client, remote_client, count_all)
|
|
|
|
for i in range(100):
|
|
count_filter = one_random_filter_please()
|
|
try:
|
|
compare_client_results(
|
|
local_client, remote_client, filter_count, count_filter=count_filter
|
|
)
|
|
except AssertionError as e:
|
|
print(f"\nFailed with filter {count_filter}")
|
|
raise e
|