Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion datafusion/src/physical_plan/repartition.rs
Original file line number Diff line number Diff line change
Expand Up @@ -536,7 +536,10 @@ mod tests {
)
.await?;

let total_rows: usize = output_partitions.iter().map(|x| x.len()).sum();
let total_rows: usize = output_partitions
.iter()
.map(|x| x.into_iter().map(|x| x.num_rows()).sum::<usize>())
.sum();

assert_eq!(8, output_partitions.len());
assert_eq!(total_rows, 8 * 50 * 3);
Expand Down