Skip to content

Commit e4f31fc

Browse files
committed
Add CSV tests
1 parent 1dd6008 commit e4f31fc

File tree

3 files changed

+25
-5
lines changed

3 files changed

+25
-5
lines changed

x-pack/plugin/esql/qa/testFixtures/src/main/resources/data/colors.csv

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
color:keyword,hex_code:keyword,rgb_vector:dense_vector
1+
color:text,hex_code:keyword,rgb_vector:dense_vector
22
maroon, #800000, [128,0,0]
33
dark red, #8B0000, [139,0,0]
44
brown, #A52A2A, [165,42,42]

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

Lines changed: 23 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ from colors metadata _score
77
| keep color, rgb_vector, _score
88
;
99

10-
color:keyword | rgb_vector:dense_vector | _score:double
10+
color:text | rgb_vector:dense_vector | _score:double
1111
green | [0.0, 128.0, 0.0] | 0.017461242154240608
1212
dark green | [0.0, 100.0, 0.0] | 0.002344448585063219
1313
forest green | [34.0, 139.0, 34.0] | 8.307271637022495E-4
@@ -29,7 +29,7 @@ from colors metadata _score
2929
| keep color, rgb_vector, _score
3030
;
3131

32-
color:keyword | rgb_vector:dense_vector | _score:double
32+
color:text | rgb_vector:dense_vector | _score:double
3333
cyan | [0.0, 255.0, 255.0] | 1.0
3434
aqua | [0.0, 255.0, 255.0] | 1.0
3535
deep sky blue | [0.0, 191.0, 255.0] | 2.6103827985934913E-4
@@ -46,7 +46,7 @@ from colors metadata _score
4646
| keep color, rgb_vector, _score
4747
;
4848

49-
color:keyword | rgb_vector:dense_vector | _score:double
49+
color:text | rgb_vector:dense_vector | _score:double
5050
pink | [255.0, 192.0, 203.0] | 1.0
5151
light pink | [255.0, 182.0, 193.0] | 0.004639764316380024
5252
peach puff | [255.0, 218.0, 185.0] | 9.59702767431736E-4
@@ -55,4 +55,24 @@ thistle | [216.0, 191.0, 216.0] | 6.913584074936807E-4
5555
wheat | [245.0, 222.0, 179.0] | 6.44357583951205E-4
5656
;
5757

58+
knnHybridSearch
59+
required_capability: knn_function
60+
61+
from colors metadata _score
62+
| where match(color, "violet") or knn(rgb_vector, [238,130,238], {"boost": 10.0, "k": 5})
63+
| sort _score desc
64+
| keep color, rgb_vector, _score
65+
;
66+
67+
color:text | rgb_vector:dense_vector| _score:double
68+
violet | [238.0, 130.0, 238.0] | 13.985496520996094
69+
blue violet | [138.0, 43.0, 226.0] | 3.132192373275757
70+
dark violet | [148.0, 0.0, 211.0] | 3.132192373275757
71+
medium violet red | [199.0, 21.0, 133.0] | 2.5798425674438477
72+
pale violet red | [219.0, 112.0, 147.0] | 2.5798425674438477
73+
orchid | [218.0, 112.0, 214.0] | 0.008259013295173645
74+
plum | [221.0, 160.0, 221.0] | 0.006797885522246361
75+
hot pink | [255.0, 105.0, 180.0] | 0.002509034238755703
76+
thistle | [216.0, 191.0, 216.0] | 0.0022455058060586452
77+
;
5878

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
{
22
"properties": {
33
"color": {
4-
"type": "keyword"
4+
"type": "text"
55
},
66
"hex_code": {
77
"type": "keyword"

0 commit comments

Comments
 (0)