Skip to content

Commit 52f057b

Browse files
committed
I give up on testing scores. You win, multiple shards on serverless.
1 parent 47b91f5 commit 52f057b

File tree

1 file changed

+49
-55
lines changed

1 file changed

+49
-55
lines changed

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

Lines changed: 49 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -71,22 +71,21 @@ required_capability: knn_function
7171
from colors metadata _score
7272
| where match(color, "violet") or knn(rgb_vector, [238,130,238], {"boost": 10.0, "k": 140})
7373
| sort _score desc
74-
| eval round_score = round(_score)
75-
| keep color, rgb_vector, round_score
74+
| keep color, rgb_vector
7675
| limit 10
7776
;
7877

79-
color:text | rgb_vector:dense_vector | round_score:double
80-
violet | [238.0, 130.0, 238.0] | 14.0
81-
blue violet | [138.0, 43.0, 226.0] | 3.0
82-
dark violet | [148.0, 0.0, 211.0] | 3.0
83-
pale violet red | [219.0, 112.0, 147.0] | 3.0
84-
medium violet red | [199.0, 21.0, 133.0] | 3.0
85-
orchid | [218.0, 112.0, 214.0] | 0.0
86-
plum | [221.0, 160.0, 221.0] | 0.0
87-
hot pink | [255.0, 105.0, 180.0] | 0.0
88-
thistle | [216.0, 191.0, 216.0] | 0.0
89-
light pink | [255.0, 182.0, 193.0] | 0.0
78+
color:text | rgb_vector:dense_vector
79+
violet | [238.0, 130.0, 238.0]
80+
blue violet | [138.0, 43.0, 226.0]
81+
dark violet | [148.0, 0.0, 211.0]
82+
pale violet red | [219.0, 112.0, 147.0]
83+
medium violet red | [199.0, 21.0, 133.0]
84+
orchid | [218.0, 112.0, 214.0]
85+
plum | [221.0, 160.0, 221.0]
86+
hot pink | [255.0, 105.0, 180.0]
87+
thistle | [216.0, 191.0, 216.0]
88+
light pink | [255.0, 182.0, 193.0]
9089
;
9190

9291
knnWithMultipleFunctions
@@ -95,14 +94,13 @@ required_capability: knn_function
9594
from colors metadata _score
9695
| where knn(rgb_vector, [128,128,0], {"k": 140}) and match(color, "olive")
9796
| sort _score desc
98-
| eval round_score = round(_score)
99-
| keep color, rgb_vector, round_score
97+
| keep color, rgb_vector
10098
;
10199

102-
color:text | rgb_vector:dense_vector | round_score:double
103-
olive | [128.0, 128.0, 0.0] | 5.0
104-
olive drab | [107.0, 142.0, 35.0] | 4.0
105-
dark olive green | [85.0, 107.0, 47.0] | 3.0
100+
color:text | rgb_vector:dense_vector
101+
olive | [128.0, 128.0, 0.0]
102+
olive drab | [107.0, 142.0, 35.0]
103+
dark olive green | [85.0, 107.0, 47.0]
106104
;
107105

108106
knnAfterKeep
@@ -200,21 +198,20 @@ from colors metadata _score
200198
| where (knn(rgb_vector, [0,255,255], {"k": 140}) or knn(rgb_vector, [128, 0, 255], {"k": 140})) and primary == true
201199
| keep color, rgb_vector, _score
202200
| sort _score desc, color asc
203-
| eval round_score = round(_score)
204201
| drop _score
205202
| limit 10
206203
;
207204

208-
color:text | rgb_vector:dense_vector | round_score:double
209-
cyan | [0.0, 255.0, 255.0] | 1.0
210-
blue | [0.0, 0.0, 255.0] | 0.0
211-
magenta | [255.0, 0.0, 255.0] | 0.0
212-
gray | [128.0, 128.0, 128.0] | 0.0
213-
white | [255.0, 255.0, 255.0] | 0.0
214-
green | [0.0, 128.0, 0.0] | 0.0
215-
black | [0.0, 0.0, 0.0] | 0.0
216-
red | [255.0, 0.0, 0.0] | 0.0
217-
yellow | [255.0, 255.0, 0.0] | 0.0
205+
color:text | rgb_vector:dense_vector
206+
cyan | [0.0, 255.0, 255.0]
207+
blue | [0.0, 0.0, 255.0]
208+
magenta | [255.0, 0.0, 255.0]
209+
gray | [128.0, 128.0, 128.0]
210+
white | [255.0, 255.0, 255.0]
211+
green | [0.0, 128.0, 0.0]
212+
black | [0.0, 0.0, 0.0]
213+
red | [255.0, 0.0, 0.0]
214+
yellow | [255.0, 255.0, 0.0]
218215
;
219216

220217
knnWithNonPushableConjunction
@@ -223,23 +220,22 @@ required_capability: knn_function
223220
from colors metadata _score
224221
| eval composed_name = locate(color, " ") > 0
225222
| where knn(rgb_vector, [128,128,0], {"k": 140}) and composed_name == false
226-
| eval round_score = round(_score)
227223
| sort _score desc, color asc
228-
| keep color, composed_name, round_score
224+
| keep color, composed_name
229225
| limit 10
230226
;
231227

232-
color:text | composed_name:boolean | round_score:double
233-
olive | false | 1.0
234-
sienna | false | 0.0
235-
chocolate | false | 0.0
236-
peru | false | 0.0
237-
brown | false | 0.0
238-
firebrick | false | 0.0
239-
chartreuse | false | 0.0
240-
gray | false | 0.0
241-
green | false | 0.0
242-
maroon | false | 0.0
228+
color:text | composed_name:boolean
229+
olive | false
230+
sienna | false
231+
chocolate | false
232+
peru | false
233+
brown | false
234+
firebrick | false
235+
chartreuse | false
236+
gray | false
237+
green | false
238+
maroon | false
243239
;
244240

245241
testKnnWithNonPushableDisjunctions
@@ -248,15 +244,14 @@ required_capability: knn_function
248244
from colors metadata _score
249245
| where knn(rgb_vector, [128,128,0], {"k": 140, "similarity": 30}) or length(color) > 17
250246
| sort _score desc, color asc
251-
| eval round_score = round(_score)
252-
| keep color, round_score
247+
| keep color
253248
;
254249

255-
color:text | round_score:double
256-
olive | 1.0
257-
light golden rod yellow | 0.0
258-
medium aqua marine | 0.0
259-
medium spring green | 0.0
250+
color:text
251+
olive
252+
light golden rod yellow
253+
medium aqua marine
254+
medium spring green
260255
;
261256

262257
testKnnWithNonPushableDisjunctionsOnComplexExpressions
@@ -265,13 +260,12 @@ required_capability: knn_function
265260
from colors metadata _score
266261
| where (knn(rgb_vector, [128,128,0], {"k": 140, "similarity": 40}) and length(color) > 5) or (knn(rgb_vector, [128,0,128], {"k": 140, "similarity": 40}) and primary == false)
267262
| sort _score desc
268-
| eval round_score = round(_score)
269-
| keep color, primary, round_score
263+
| keep color, primary
270264
;
271265

272-
color:text | primary:boolean | round_score:double
273-
purple | false | 1.0
274-
dark magenta | false | 0.0
266+
color:text | primary:boolean
267+
purple | false
268+
dark magenta | false
275269
;
276270

277271
testKnnInStatsNonPushable

0 commit comments

Comments
 (0)