Skip to content

Commit 5bf0bf1

Browse files
committed
tests: add test from lookup_from
1 parent aab4cdb commit 5bf0bf1

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/congruence_tests/test_query.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,18 @@ def dense_query_group_with_lookup(self, client: QdrantBase) -> GroupsResult:
240240
with_lookup=SECONDARY_COLLECTION_NAME,
241241
)
242242

243+
def dense_query_group_with_lookup_from(self, client: QdrantBase) -> GroupsResult:
244+
return client.query_points_groups(
245+
collection_name=COLLECTION_NAME,
246+
query=1,
247+
using="text",
248+
group_by=self.group_by,
249+
group_size=self.group_size,
250+
limit=self.limit,
251+
with_payload=models.PayloadSelectorInclude(include=[self.group_by]),
252+
lookup_from=models.LookupLocation(collection=SECONDARY_COLLECTION_NAME, vector="text"),
253+
)
254+
243255
def filter_dense_query_group(
244256
self, client: QdrantBase, query_filter: models.Filter
245257
) -> GroupsResult:
@@ -1690,6 +1702,10 @@ def test_query_group():
16901702
compare_clients_results(
16911703
local_client, http_client, grpc_client, searcher.dense_query_group_with_lookup
16921704
)
1705+
compare_clients_results(
1706+
local_client, http_client, grpc_client, searcher.dense_query_group_with_lookup_from
1707+
)
1708+
16931709
compare_clients_results(
16941710
local_client, http_client, grpc_client, searcher.dense_queries_rescore_group
16951711
)

0 commit comments

Comments
 (0)