Add CSV tests

Author: carlosdelest

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

@@ -209,13 +209,122 @@ required_capability: knn_function
 
 from colors metadata _score
 | eval composed_name = locate(color, " ") > 0 
-| where knn(rgb_vector, [128,128,0]) and composed_name == false 
-| keep color, composed_name, _score
+| where knn(rgb_vector, [128,128,0]) and composed_name == false
+| eval round_score = round(_score, 4)
+| keep color, composed_name, round_score
 ;
 
-color:text    | composed_name: boolean | _score: double
-olive            | false                  | 1.0                  
-sienna           | false                  | 2.9301168979145586E-4
-chocolate        | false                  | 1.4101568376645446E-4
-peru             | false                  | 1.3746823242399842E-4      
+color:text | composed_name:boolean | round_score:double
+olive      | false                 | 1.0               
+sienna     | false                 | 3.0E-4            
+chocolate  | false                 | 1.0E-4            
+peru       | false                 | 1.0E-4    
 ;
+
+testKnnWithNonPushableDisjunctions
+required_capability: knn_function
+
+from colors metadata _score 
+| where knn(rgb_vector, [128,128,0], {"k": 5}) or length(color) > 17 
+| sort _score desc
+| eval round_score = round(_score, 4)
+| keep color, round_score 
+;
+
+color:text              | round_score: double  
+olive                   | 1.0            
+olive drab              | 0.0014         
+dark olive green        | 4.0E-4         
+dark golden rod         | 3.0E-4         
+sienna                  | 3.0E-4         
+medium aqua marine      | 0.0            
+medium spring green     | 0.0            
+light golden rod yellow | 0.0            
+;
+
+testKnnWithNonPushableDisjunctionsOnComplexExpressions
+required_capability: knn_function
+
+from colors metadata _score 
+| where (knn(rgb_vector, [128,128,0]) and length(color) > 12) or (knn(rgb_vector, [128,0,128]) and primary == false) 
+| sort _score desc 
+| eval round_score = round(_score, 4) 
+| keep color, primary, round_score
+;
+
+color: text       | primary: boolean   | round_score: double  
+purple            | false              | 1.0            
+dark magenta      | false              | 0.0045         
+dark olive green  | false              | 4.0E-4         
+indigo            | false              | 4.0E-4         
+dark golden rod   | false              | 3.0E-4         
+dim gray          | false              | 3.0E-4         
+dark slate blue   | false              | 2.0E-4         
+medium violet red | false              | 2.0E-4         
+dark orchid       | false              | 1.0E-4         
+dark violet       | false              | 1.0E-4         
+brown             | false              | 1.0E-4         
+blue violet       | false              | 1.0E-4         
+;
+
+testKnnInStatsNonPushable
+required_capability: knn_function
+
+from colors 
+| where length(color) < 10 
+| stats c = count(*) where knn(rgb_vector, [128,128,255], {"k": 40})
+;
+
+c: long 
+11      
+;
+
+
+testKnnInStatsPushableAndNonPushable
+required_capability: knn_function
+required_capability: full_text_functions_in_stats_where
+
+from colors metadata _score 
+| stats c = count(*) where (knn(rgb_vector, [0,255,255], {"k": 40}) or knn(rgb_vector, [0, 0, 255])) and knn(rgb_vector, [0, 255, 0], {"k": 40})
+;
+
+c:long
+21
+;
+
+testKnnInStatsWithGrouping
+from colors 
+| where length(color) < 10 
+| stats c = count(*) where knn(rgb_vector, [128,128,255], {"k": 40}) by primary
+;
+
+c: long       | primary: boolean    
+9             | false          
+2             | true           
+;
+
+testKnnInStatsPushable
+required_capability: knn_function
+required_capability: full_text_functions_in_stats_where
+
+from colors
+| stats c = count(*) where knn(rgb_vector, [128,128,255], {"k": 40})
+;
+
+# No surprises, gets the number of top k
+c:long
+40
+;
+
+testKnnInStatsWithNonPushableDisjunctions
+required_capability: knn_function
+required_capability: full_text_functions_in_stats_where
+
+FROM colors
+| STATS c = count(*) where knn(rgb_vector, [128,128,0], {"k": 5}) or length(color) > 17 
+;
+
+c:long
+8
+;
+