Skip to content

Commit ab1943e

Browse files
committed
Add CSV tests
1 parent bdfc1c2 commit ab1943e

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

x-pack/plugin/esql/qa/testFixtures/src/main/resources/knn-function.csv-spec

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ required_capability: knn_function
77

88
// tag::knn-function[]
99
from colors metadata _score
10-
| where knn(rgb_vector, [0, 120, 0])
10+
| where knn(rgb_vector, [0, 120, 0], 10)
1111
| sort _score desc, color asc
1212
// end::knn-function[]
1313
| keep color, rgb_vector
@@ -34,7 +34,7 @@ required_capability: knn_function
3434

3535
// tag::knn-function-options[]
3636
from colors metadata _score
37-
| where knn(rgb_vector, [0,255,255], {"k": 4})
37+
| where knn(rgb_vector, [0,255,255], 140)
3838
| sort _score desc, color asc
3939
// end::knn-function-options[]
4040
| keep color, rgb_vector
@@ -53,7 +53,7 @@ knnSearchWithSimilarityOption-Ignore
5353
required_capability: knn_function
5454

5555
from colors metadata _score
56-
| where knn(rgb_vector, [255,192,203], {"k": 140, "similarity": 40})
56+
| where knn(rgb_vector, [255,192,203], 140, {"similarity": 40})
5757
| sort _score desc, color asc
5858
| keep color, rgb_vector
5959
;
@@ -70,7 +70,7 @@ knnHybridSearch
7070
required_capability: knn_function
7171

7272
from colors metadata _score
73-
| where match(color, "blue") or knn(rgb_vector, [65,105,225], {"k": 140})
73+
| where match(color, "blue") or knn(rgb_vector, [65,105,225], 140)
7474
| where primary == true
7575
| sort _score desc, color asc
7676
| keep color, rgb_vector
@@ -93,7 +93,7 @@ knnWithMultipleFunctions
9393
required_capability: knn_function
9494

9595
from colors metadata _score
96-
| where knn(rgb_vector, [128,128,0], {"k": 140}) and match(color, "olive")
96+
| where knn(rgb_vector, [128,128,0], 140) and match(color, "olive")
9797
| sort _score desc, color asc
9898
| keep color, rgb_vector
9999
;
@@ -107,7 +107,7 @@ required_capability: knn_function
107107

108108
from colors metadata _score
109109
| keep rgb_vector, color, _score
110-
| where knn(rgb_vector, [128,255,0], {"k": 140})
110+
| where knn(rgb_vector, [128,255,0], 140)
111111
| sort _score desc, color asc
112112
| keep rgb_vector
113113
| limit 5
@@ -126,7 +126,7 @@ required_capability: knn_function
126126

127127
from colors metadata _score
128128
| drop primary
129-
| where knn(rgb_vector, [128,250,0], {"k": 140})
129+
| where knn(rgb_vector, [128,250,0], 140)
130130
| sort _score desc, color asc
131131
| keep color, rgb_vector
132132
| limit 5
@@ -145,7 +145,7 @@ required_capability: knn_function
145145

146146
from colors metadata _score
147147
| eval composed_name = locate(color, " ") > 0
148-
| where knn(rgb_vector, [128,128,0], {"k": 140})
148+
| where knn(rgb_vector, [128,128,0], 140)
149149
| sort _score desc, color asc
150150
| keep color, composed_name
151151
| limit 5
@@ -164,7 +164,7 @@ required_capability: knn_function
164164

165165
# TODO We need kNN prefiltering here so we get more candidates that pass the filter
166166
from colors metadata _score
167-
| where knn(rgb_vector, [255,255,238], {"k": 140}) and hex_code like "#FFF*"
167+
| where knn(rgb_vector, [255,255,238], 140) and hex_code like "#FFF*"
168168
| sort _score desc, color asc
169169
| keep color, hex_code, rgb_vector
170170
| limit 10
@@ -185,7 +185,7 @@ required_capability: knn_function
185185

186186
# TODO We need kNN prefiltering here so we get more candidates that pass the filter
187187
from colors metadata _score
188-
| where (knn(rgb_vector, [0,255,255], {"k": 140}) or knn(rgb_vector, [128, 0, 255], {"k": 140})) and primary == true
188+
| where (knn(rgb_vector, [0,255,255], 140) or knn(rgb_vector, [128, 0, 255], 140)) and primary == true
189189
| keep color, rgb_vector, _score
190190
| sort _score desc, color asc
191191
| drop _score
@@ -209,7 +209,7 @@ required_capability: knn_function
209209

210210
from colors metadata _score
211211
| eval composed_name = locate(color, " ") > 0
212-
| where knn(rgb_vector, [128,128,0], {"k": 140}) and composed_name == false
212+
| where knn(rgb_vector, [128,128,0], 140) and composed_name == false
213213
| sort _score desc, color asc
214214
| keep color, composed_name
215215
| limit 10
@@ -233,7 +233,7 @@ testKnnWithNonPushableDisjunctions-Ignore
233233
required_capability: knn_function
234234

235235
from colors metadata _score
236-
| where knn(rgb_vector, [128,128,0], {"k": 140, "similarity": 30}) or length(color) > 10
236+
| where knn(rgb_vector, [128,128,0], 140, {"similarity": 30}) or length(color) > 10
237237
| sort _score desc, color asc
238238
| keep color
239239
;
@@ -250,7 +250,7 @@ testKnnWithNonPushableDisjunctionsOnComplexExpressions-Ignore
250250
required_capability: knn_function
251251

252252
from colors metadata _score
253-
| where (knn(rgb_vector, [128,128,0], {"k": 140, "similarity": 70}) and length(color) < 10) or (knn(rgb_vector, [128,0,128], {"k": 140, "similarity": 60}) and primary == false)
253+
| where (knn(rgb_vector, [128,128,0], 140, {"similarity": 70}) and length(color) < 10) or (knn(rgb_vector, [128,0,128], 140, {"similarity": 60}) and primary == false)
254254
| sort _score desc, color asc
255255
| keep color, primary
256256
;
@@ -266,7 +266,7 @@ required_capability: knn_function
266266

267267
from colors
268268
| where length(color) < 10
269-
| stats c = count(*) where knn(rgb_vector, [128,128,255], {"k": 140})
269+
| stats c = count(*) where knn(rgb_vector, [128,128,255], 140)
270270
;
271271

272272
c: long
@@ -279,7 +279,7 @@ required_capability: full_text_functions_in_stats_where
279279

280280
from colors
281281
| where length(color) < 10
282-
| stats c = count(*) where knn(rgb_vector, [128,128,255], {"k": 140}) by primary
282+
| stats c = count(*) where knn(rgb_vector, [128,128,255], 140) by primary
283283
;
284284

285285
c: long | primary: boolean

0 commit comments

Comments
 (0)