Skip to content

Commit f4c6cfa

Browse files
fix: compatibility with #364
1 parent cc1ef1f commit f4c6cfa

File tree

1 file changed

+6
-11
lines changed

1 file changed

+6
-11
lines changed

crates/iceberg/src/writer/base_writer/equality_delete_writer.rs

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,6 @@ impl FieldProjector {
232232
mod 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
@@ -295,7 +290,7 @@ mod test {
295290
.sum::<i64>() as u64
296291
);
297292

298-
assert_eq!(data_file.file_size_in_bytes, file_size as u64);
293+
assert_eq!(data_file.file_size_in_bytes, input_content.len() as u64);
299294

300295
assert_eq!(data_file.column_sizes.len(), expect_column_num);
301296

0 commit comments

Comments
 (0)