Skip to content

Commit 521d119

Browse files
committed
added some yaml tests only as parsing validation
1 parent 65389c8 commit 521d119

File tree

3 files changed

+67
-0
lines changed

3 files changed

+67
-0
lines changed

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.retrievers/20_knn_retriever.yml

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,27 @@ setup:
5353
- match: {hits.hits.1._id: "3"}
5454
- match: {hits.hits.1.fields.name.0: "rabbit.jpg"}
5555

56+
---
57+
"kNN retrieve with visit_percentage":
58+
- do:
59+
search:
60+
index: index1
61+
body:
62+
fields: [ "name" ]
63+
retriever:
64+
knn:
65+
field: vector
66+
query_vector: [2, 2, 2, 2, 3]
67+
k: 2
68+
num_candidates: 3
69+
visit_percentage: 1.0
70+
71+
- match: {hits.hits.0._id: "2"}
72+
- match: {hits.hits.0.fields.name.0: "moose.jpg"}
73+
74+
- match: {hits.hits.1._id: "3"}
75+
- match: {hits.hits.1.fields.name.0: "rabbit.jpg"}
76+
5677
---
5778
"kNN retriever with filter":
5879
- do:

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/135_knn_query_nested_search_ivf.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,3 +191,27 @@ setup:
191191
- match: {hits.hits.0._id: "3"}
192192
- match: {hits.hits.0.fields.name.0: "rabbit.jpg"}
193193
- match: { hits.hits.0.inner_hits.nested.hits.hits.0.fields.nested.0.paragraph_id.0: "0" }
194+
195+
---
196+
"nested kNN search works with visit_percentage":
197+
- do:
198+
search:
199+
index: test
200+
body:
201+
fields: [ "name" ]
202+
query:
203+
nested:
204+
path: nested
205+
query:
206+
knn:
207+
field: nested.vector
208+
query_vector: [-0.5, 90, -10, 14.8, -156]
209+
num_candidates: 3
210+
visit_percentage: 1.0
211+
- match: {hits.total.value: 3}
212+
213+
- match: {hits.hits.0._id: "2"}
214+
- match: {hits.hits.0.fields.name.0: "moose.jpg"}
215+
216+
- match: {hits.hits.1._id: "3"}
217+
- match: {hits.hits.1.fields.name.0: "rabbit.jpg"}

rest-api-spec/src/yamlRestTest/resources/rest-api-spec/test/search.vectors/46_knn_search_bbq_ivf.yml

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,28 @@ setup:
106106
- match: { hits.hits.1._id: "3" }
107107
- match: { hits.hits.2._id: "2" }
108108
---
109+
"Test knn search with visit_percentage":
110+
- do:
111+
search:
112+
index: bbq_disk
113+
body:
114+
knn:
115+
field: vector
116+
query_vector: [0.128, 0.067, -0.08 , 0.395, -0.11 , -0.259, 0.473, -0.393,
117+
0.292, 0.571, -0.491, 0.444, -0.288, 0.198, -0.343, 0.015,
118+
0.232, 0.088, 0.228, 0.151, -0.136, 0.236, -0.273, -0.259,
119+
-0.217, 0.359, -0.207, 0.352, -0.142, 0.192, -0.061, -0.17 ,
120+
-0.343, 0.189, -0.221, 0.32 , -0.301, -0.1 , 0.005, 0.232,
121+
-0.344, 0.136, 0.252, 0.157, -0.13 , -0.244, 0.193, -0.034,
122+
-0.12 , -0.193, -0.102, 0.252, -0.185, -0.167, -0.575, 0.582,
123+
-0.426, 0.983, 0.212, 0.204, 0.03 , -0.276, -0.425, -0.158]
124+
k: 3
125+
visit_percentage: 1.0
126+
127+
- match: { hits.hits.0._id: "1" }
128+
- match: { hits.hits.1._id: "3" }
129+
- match: { hits.hits.2._id: "2" }
130+
---
109131
"Vector rescoring has same scoring as exact search for kNN section":
110132
- skip:
111133
features: "headers"

0 commit comments

Comments
 (0)