Skip to content

Commit 6d6b81f

Browse files
committed
deprecate: replace search with query points in test delete points
1 parent 20d6074 commit 6d6b81f

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

tests/congruence_tests/test_delete_points.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,14 @@ def test_delete_points(local_client, remote_client):
2121
compare_client_results(
2222
local_client,
2323
remote_client,
24-
lambda c: c.search(COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)),
24+
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="image").points,
2525
)
2626

2727
found_ids = [
2828
scored_point.id
29-
for scored_point in local_client.search(
30-
COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)
31-
)
29+
for scored_point in local_client.query_points(
30+
COLLECTION_NAME, query=vector, using="image"
31+
).points
3232
]
3333

3434
local_client.delete(COLLECTION_NAME, found_ids)
@@ -39,10 +39,10 @@ def test_delete_points(local_client, remote_client):
3939
compare_client_results(
4040
local_client,
4141
remote_client,
42-
lambda c: c.search(COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)),
42+
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="image").points,
4343
)
4444

45-
#delete non-existent points
45+
# delete non-existent points
4646
local_client.delete(COLLECTION_NAME, found_ids)
4747
remote_client.delete(COLLECTION_NAME, found_ids)
4848

@@ -51,7 +51,7 @@ def test_delete_points(local_client, remote_client):
5151
compare_client_results(
5252
local_client,
5353
remote_client,
54-
lambda c: c.search(COLLECTION_NAME, query_vector=NamedVector(name="image", vector=vector)),
54+
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="image").points,
5555
)
5656

5757

@@ -71,16 +71,18 @@ def test_delete_sparse_points():
7171
compare_client_results(
7272
local_client,
7373
remote_client,
74-
lambda c: c.search(
75-
COLLECTION_NAME, query_vector=NamedSparseVector(name="sparse-image", vector=vector)
76-
),
74+
lambda c: c.query_points(
75+
COLLECTION_NAME,
76+
query=vector,
77+
using="sparse-image",
78+
).points,
7779
)
7880

7981
found_ids = [
8082
scored_point.id
81-
for scored_point in local_client.search(
82-
COLLECTION_NAME, query_vector=NamedSparseVector(name="sparse-image", vector=vector)
83-
)
83+
for scored_point in local_client.query_points(
84+
COLLECTION_NAME, query=vector, using="sparse-image"
85+
).points
8486
]
8587

8688
local_client.delete(COLLECTION_NAME, found_ids)
@@ -91,7 +93,5 @@ def test_delete_sparse_points():
9193
compare_client_results(
9294
local_client,
9395
remote_client,
94-
lambda c: c.search(
95-
COLLECTION_NAME, query_vector=NamedSparseVector(name="sparse-image", vector=vector)
96-
),
96+
lambda c: c.query_points(COLLECTION_NAME, query=vector, using="sparse-image").points,
9797
)

0 commit comments

Comments
 (0)