@@ -20,24 +20,6 @@ use datafusion::scalar::ScalarValue;
20
20
use datafusion:: test_util:: scan_empty;
21
21
use datafusion_common:: cast:: as_float64_array;
22
22
23
- #[ tokio:: test]
24
- async fn csv_query_avg_multi_batch ( ) -> Result < ( ) > {
25
- let ctx = SessionContext :: new ( ) ;
26
- register_aggregate_csv ( & ctx) . await ?;
27
- let sql = "SELECT avg(c12) FROM aggregate_test_100" ;
28
- let dataframe = ctx. sql ( sql) . await . unwrap ( ) ;
29
- let results = dataframe. collect ( ) . await . unwrap ( ) ;
30
- let batch = & results[ 0 ] ;
31
- let column = batch. column ( 0 ) ;
32
- let array = as_float64_array ( column) ?;
33
- let actual = array. value ( 0 ) ;
34
- let expected = 0.5089725 ;
35
- // Due to float number's accuracy, different batch size will lead to different
36
- // answers.
37
- assert ! ( ( expected - actual) . abs( ) < 0.01 ) ;
38
- Ok ( ( ) )
39
- }
40
-
41
23
#[ tokio:: test]
42
24
#[ ignore] // https://github.com/apache/arrow-datafusion/issues/3353
43
25
async fn csv_query_approx_count ( ) -> Result < ( ) > {
@@ -120,60 +102,6 @@ async fn csv_query_approx_percentile_cont_with_histogram_bins() -> Result<()> {
120
102
Ok ( ( ) )
121
103
}
122
104
123
- #[ tokio:: test]
124
- async fn csv_query_array_agg ( ) -> Result < ( ) > {
125
- let ctx = SessionContext :: new ( ) ;
126
- register_aggregate_csv ( & ctx) . await ?;
127
- let sql =
128
- "SELECT array_agg(c13) FROM (SELECT * FROM aggregate_test_100 ORDER BY c13 LIMIT 2) test" ;
129
- let actual = execute_to_batches ( & ctx, sql) . await ;
130
- let expected = vec ! [
131
- "+------------------------------------------------------------------+" ,
132
- "| ARRAYAGG(test.c13) |" ,
133
- "+------------------------------------------------------------------+" ,
134
- "| [0VVIHzxWtNOFLtnhjHEKjXaJOSLJfm, 0keZ5G8BffGwgF2RwQD59TFzMStxCB] |" ,
135
- "+------------------------------------------------------------------+" ,
136
- ] ;
137
- assert_batches_eq ! ( expected, & actual) ;
138
- Ok ( ( ) )
139
- }
140
-
141
- #[ tokio:: test]
142
- async fn csv_query_array_agg_empty ( ) -> Result < ( ) > {
143
- let ctx = SessionContext :: new ( ) ;
144
- register_aggregate_csv ( & ctx) . await ?;
145
- let sql =
146
- "SELECT array_agg(c13) FROM (SELECT * FROM aggregate_test_100 LIMIT 0) test" ;
147
- let actual = execute_to_batches ( & ctx, sql) . await ;
148
- let expected = vec ! [
149
- "+--------------------+" ,
150
- "| ARRAYAGG(test.c13) |" ,
151
- "+--------------------+" ,
152
- "| [] |" ,
153
- "+--------------------+" ,
154
- ] ;
155
- assert_batches_eq ! ( expected, & actual) ;
156
- Ok ( ( ) )
157
- }
158
-
159
- #[ tokio:: test]
160
- async fn csv_query_array_agg_one ( ) -> Result < ( ) > {
161
- let ctx = SessionContext :: new ( ) ;
162
- register_aggregate_csv ( & ctx) . await ?;
163
- let sql =
164
- "SELECT array_agg(c13) FROM (SELECT * FROM aggregate_test_100 ORDER BY c13 LIMIT 1) test" ;
165
- let actual = execute_to_batches ( & ctx, sql) . await ;
166
- let expected = vec ! [
167
- "+----------------------------------+" ,
168
- "| ARRAYAGG(test.c13) |" ,
169
- "+----------------------------------+" ,
170
- "| [0VVIHzxWtNOFLtnhjHEKjXaJOSLJfm] |" ,
171
- "+----------------------------------+" ,
172
- ] ;
173
- assert_batches_eq ! ( expected, & actual) ;
174
- Ok ( ( ) )
175
- }
176
-
177
105
#[ tokio:: test]
178
106
async fn csv_query_array_agg_unsupported ( ) -> Result < ( ) > {
179
107
let ctx = SessionContext :: new ( ) ;
0 commit comments