File tree Expand file tree Collapse file tree 1 file changed +5
-10
lines changed
crates/iceberg/src/writer/base_writer Expand file tree Collapse file tree 1 file changed +5
-10
lines changed Original file line number Diff line number Diff line change @@ -232,7 +232,6 @@ impl FieldProjector {
232232mod test {
233233 use arrow_select:: concat:: concat_batches;
234234 use bytes:: Bytes ;
235- use futures:: AsyncReadExt ;
236235 use itertools:: Itertools ;
237236 use std:: { collections:: HashMap , sync:: Arc } ;
238237
@@ -266,15 +265,11 @@ mod test {
266265 assert_eq ! ( data_file. file_format, DataFileFormat :: Parquet ) ;
267266
268267 // read the written file
269- let mut input_file = file_io
270- . new_input ( data_file. file_path . clone ( ) )
271- . unwrap ( )
272- . reader ( )
273- . await
274- . unwrap ( ) ;
275- let mut res = vec ! [ ] ;
276- let file_size = input_file. read_to_end ( & mut res) . await . unwrap ( ) ;
277- let reader_builder = ParquetRecordBatchReaderBuilder :: try_new ( Bytes :: from ( res) ) . unwrap ( ) ;
268+ let input_file = file_io. new_input ( data_file. file_path . clone ( ) ) . unwrap ( ) ;
269+ // read the written file
270+ let input_content = input_file. read ( ) . await . unwrap ( ) ;
271+ let reader_builder =
272+ ParquetRecordBatchReaderBuilder :: try_new ( input_content. clone ( ) ) . unwrap ( ) ;
278273 let metadata = reader_builder. metadata ( ) . clone ( ) ;
279274
280275 // check data
You can’t perform that action at this time.
0 commit comments