Skip to content

Commit e7452dd

Browse files
committed
Add CSV tests
1 parent 26f48e7 commit e7452dd

File tree

1 file changed

+24
-20
lines changed

1 file changed

+24
-20
lines changed

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

Lines changed: 24 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,41 +1,45 @@
11
knnSearch
22
required_capability: knn_function
33

4+
// tag::knn-function[]
45
from colors metadata _score
56
| where knn(rgb_vector, [0, 120, 0])
67
| sort _score desc
7-
| eval round_score = round(_score, 4)
8-
| keep color, rgb_vector, round_score
8+
// end::knn-function[]
9+
| keep color, rgb_vector
910
;
1011

11-
color:text | rgb_vector:dense_vector | round_score:double
12-
green | [0.0, 128.0, 0.0] | 0.0127
13-
dark green | [0.0, 100.0, 0.0] | 0.0027
14-
forest green | [34.0, 139.0, 34.0] | 0.0023
15-
dark olive green | [85.0, 107.0, 47.0] | 3.0E-4
16-
sea green | [46.0, 139.0, 87.0] | 2.0E-4
17-
dark slate gray | [47.0, 79.0, 79.0] | 2.0E-4
18-
olive drab | [107.0, 142.0, 35.0] | 1.0E-4
19-
lime green | [50.0, 205.0, 50.0] | 1.0E-4
20-
black | [0.0, 0.0, 0.0] | 1.0E-4
21-
olive | [128.0, 128.0, 0.0] | 1.0E-4
12+
// tag::knn-function-result[]
13+
color:text | rgb_vector:dense_vector
14+
green | [0.0, 128.0, 0.0]
15+
dark green | [0.0, 100.0, 0.0]
16+
forest green | [34.0, 139.0, 34.0]
17+
dark olive green | [85.0, 107.0, 47.0]
18+
sea green | [46.0, 139.0, 87.0]
19+
dark slate gray | [47.0, 79.0, 79.0]
20+
olive drab | [107.0, 142.0, 35.0]
21+
lime green | [50.0, 205.0, 50.0]
22+
black | [0.0, 0.0, 0.0]
23+
olive | [128.0, 128.0, 0.0]
24+
// end::knn-function-result[]
2225
;
2326

2427
knnSearchWithKOption
2528
required_capability: knn_function
2629

30+
// tag::knn-function-options[]
2731
from colors metadata _score
2832
| where knn(rgb_vector, [0,255,255], {"k": 4})
2933
| sort _score desc
30-
| eval round_score = round(_score, 4)
31-
| keep color, rgb_vector, round_score
34+
// end::knn-function-options[]
35+
| keep color, rgb_vector
3236
;
3337

34-
color:text | rgb_vector:dense_vector | round_score:double
35-
cyan | [0.0, 255.0, 255.0] | 1.0
36-
deep sky blue | [0.0, 191.0, 255.0] | 3.0E-4
37-
dark turquoise | [0.0, 206.0, 209.0] | 3.0E-4
38-
turquoise | [64.0, 224.0, 208.0] | 3.0E-4
38+
color:text | rgb_vector:dense_vector
39+
cyan | [0.0, 255.0, 255.0]
40+
deep sky blue | [0.0, 191.0, 255.0]
41+
dark turquoise | [0.0, 206.0, 209.0]
42+
turquoise | [64.0, 224.0, 208.0]
3943
;
4044

4145
knnSearchWithSimilarityOption

0 commit comments

Comments
 (0)