Skip to content

Commit 78aa6d0

Browse files
committed
Remove quantization for less brittle tests
1 parent d824faa commit 78aa6d0

File tree

2 files changed

+29
-29
lines changed

2 files changed

+29
-29
lines changed

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

Lines changed: 22 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -8,23 +8,23 @@ required_capability: knn_function
88
// tag::knn-function[]
99
from colors metadata _score
1010
| where knn(rgb_vector, [0, 120, 0])
11-
| sort _score desc
11+
| sort _score desc, color asc
1212
// end::knn-function[]
1313
| keep color, rgb_vector
1414
;
1515

1616
// tag::knn-function-result[]
1717
color:text | rgb_vector:dense_vector
18-
green | [0.0, 128.0, 0.0]
19-
dark green | [0.0, 100.0, 0.0]
20-
forest green | [34.0, 139.0, 34.0]
21-
dark olive green | [85.0, 107.0, 47.0]
22-
sea green | [46.0, 139.0, 87.0]
23-
dark slate gray | [47.0, 79.0, 79.0]
24-
olive drab | [107.0, 142.0, 35.0]
25-
lime green | [50.0, 205.0, 50.0]
26-
black | [0.0, 0.0, 0.0]
27-
olive | [128.0, 128.0, 0.0]
18+
green | [0.0, 128.0, 0.0]
19+
dark green | [0.0, 100.0, 0.0]
20+
forest green | [34.0, 139.0, 34.0]
21+
dark olive green | [85.0, 107.0, 47.0]
22+
sea green | [46.0, 139.0, 87.0]
23+
dark slate gray | [47.0, 79.0, 79.0]
24+
lime green | [50.0, 205.0, 50.0]
25+
olive drab | [107.0, 142.0, 35.0]
26+
black | [0.0, 0.0, 0.0]
27+
olive | [128.0, 128.0, 0.0]
2828
// end::knn-function-result[]
2929
;
3030

@@ -52,17 +52,16 @@ required_capability: knn_function
5252

5353
from colors metadata _score
5454
| where knn(rgb_vector, [255,192,203], {"k": 140, "similarity": 40})
55-
| sort _score desc
55+
| sort _score desc, color asc
5656
| keep color, rgb_vector
5757
;
5858

5959
color:text | rgb_vector:dense_vector
60-
pink | [255.0, 192.0, 203.0]
61-
light pink | [255.0, 182.0, 193.0]
62-
peach puff | [255.0, 218.0, 185.0]
63-
bisque | [255.0, 228.0, 196.0]
64-
thistle | [216.0, 191.0, 216.0]
65-
wheat | [245.0, 222.0, 179.0]
60+
pink | [255.0, 192.0, 203.0]
61+
light pink | [255.0, 182.0, 193.0]
62+
peach puff | [255.0, 218.0, 185.0]
63+
bisque | [255.0, 228.0, 196.0]
64+
wheat | [245.0, 222.0, 179.0]
6665
;
6766

6867
knnHybridSearch
@@ -157,14 +156,14 @@ from colors metadata _score
157156
color:text | composed_name:boolean
158157
olive | false
159158
olive drab | true
160-
dark olive green | true
161159
dark golden rod | true
162-
sienna | false
163160
saddle brown | true
164-
dim gray | true
161+
dark olive green | true
162+
sienna | false
165163
chocolate | false
166164
yellow green | true
167165
peru | false
166+
forest green | true
168167
;
169168

170169
knnWithConjunction
@@ -234,11 +233,11 @@ sienna | false | 0.0
234233
chocolate | false | 0.0
235234
peru | false | 0.0
236235
brown | false | 0.0
236+
firebrick | false | 0.0
237+
chartreuse | false | 0.0
237238
gray | false | 0.0
238239
green | false | 0.0
239240
maroon | false | 0.0
240-
firebrick | false | 0.0
241-
chartreuse | false | 0.0
242241
;
243242

244243
testKnnWithNonPushableDisjunctions
@@ -253,7 +252,6 @@ from colors metadata _score
253252

254253
color:text | round_score:double
255254
olive | 1.0
256-
olive drab | 0.0
257255
light golden rod yellow | 0.0
258256
medium aqua marine | 0.0
259257
medium spring green | 0.0
@@ -272,7 +270,6 @@ from colors metadata _score
272270
color:text | primary:boolean | round_score:double
273271
purple | false | 1.0
274272
dark magenta | false | 0.0
275-
olive drab | false | 0.0
276273
;
277274

278275
testKnnInStatsNonPushable

x-pack/plugin/esql/qa/testFixtures/src/main/resources/mapping-colors.json

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,15 @@
66
"hex_code": {
77
"type": "keyword"
88
},
9-
"rgb_vector": {
10-
"type": "dense_vector",
11-
"similarity": "l2_norm"
12-
},
139
"primary": {
1410
"type": "boolean"
11+
},
12+
"rgb_vector": {
13+
"type": "dense_vector",
14+
"similarity": "l2_norm",
15+
"index_options": {
16+
"type": "hnsw"
17+
}
1518
}
1619
}
1720
}

0 commit comments

Comments
 (0)