@@ -24,15 +24,16 @@ use arrow_schema::{DataType, Field, FieldRef, Schema, SchemaRef};
24
24
use bytes:: { BufMut , BytesMut } ;
25
25
use datafusion:: assert_batches_eq;
26
26
use datafusion:: common:: Result ;
27
+ use datafusion:: datasource:: listing:: PartitionedFile ;
27
28
use datafusion:: datasource:: listing:: { ListingTable , ListingTableConfig } ;
28
- use datafusion:: datasource:: physical_plan:: { ArrowSource , CsvSource , FileSource , JsonSource , ParquetSource } ;
29
- use datafusion_physical_plan:: ExecutionPlan ;
29
+ use datafusion:: datasource:: physical_plan:: {
30
+ ArrowSource , CsvSource , FileSource , JsonSource , ParquetSource ,
31
+ } ;
30
32
use datafusion:: prelude:: { SessionConfig , SessionContext } ;
31
33
use datafusion_common:: tree_node:: { Transformed , TransformedResult , TreeNode } ;
32
34
use datafusion_common:: DataFusionError ;
33
35
use datafusion_common:: { ColumnStatistics , ScalarValue } ;
34
36
use datafusion_datasource:: file_scan_config:: FileScanConfigBuilder ;
35
- use datafusion:: datasource:: listing:: PartitionedFile ;
36
37
use datafusion_datasource:: schema_adapter:: {
37
38
DefaultSchemaAdapterFactory , SchemaAdapter , SchemaAdapterFactory , SchemaMapper ,
38
39
} ;
@@ -44,6 +45,7 @@ use datafusion_physical_expr::schema_rewriter::{
44
45
DefaultPhysicalExprAdapterFactory , PhysicalExprAdapter , PhysicalExprAdapterFactory ,
45
46
} ;
46
47
use datafusion_physical_expr:: { DefaultPhysicalExprAdapter , PhysicalExpr } ;
48
+ use datafusion_physical_plan:: ExecutionPlan ;
47
49
use itertools:: Itertools ;
48
50
use object_store:: { memory:: InMemory , path:: Path , ObjectStore } ;
49
51
use parquet:: arrow:: ArrowWriter ;
@@ -386,7 +388,6 @@ async fn test_custom_schema_adapter_and_custom_expression_adapter() {
386
388
#[ derive( Debug , PartialEq ) ]
387
389
struct UppercaseAdapterFactory { }
388
390
389
-
390
391
impl SchemaAdapterFactory for UppercaseAdapterFactory {
391
392
fn create (
392
393
& self ,
@@ -397,10 +398,8 @@ impl SchemaAdapterFactory for UppercaseAdapterFactory {
397
398
table_schema : projected_table_schema,
398
399
} )
399
400
}
400
-
401
401
}
402
402
403
-
404
403
/// Schema adapter that transforms column names to uppercase
405
404
#[ derive( Debug ) ]
406
405
struct UppercaseAdapter {
0 commit comments