[arrow-datafusion] branch master updated: fix calculate in many_to_many_hash_partition test. (#1463)

[dh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

dheres pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow-datafusion.git


The following commit(s) were added to refs/heads/master by this push:
     new 35d65fc  fix calculate in many_to_many_hash_partition test. (#1463)
35d65fc is described below

commit 35d65fc37cf2319c4dbec32102adb10502847462
Author: Yang <37...@users.noreply.github.com>
AuthorDate: Sat Dec 18 22:41:36 2021 +0800

    fix calculate in many_to_many_hash_partition test. (#1463)
    
    * fix calculate in many_to_many_hash_partition test.
    
    * fix Clippy Lints
---
 datafusion/src/physical_plan/repartition.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/datafusion/src/physical_plan/repartition.rs b/datafusion/src/physical_plan/repartition.rs
index 55dab6c..3cc7d54 100644
--- a/datafusion/src/physical_plan/repartition.rs
+++ b/datafusion/src/physical_plan/repartition.rs
@@ -576,7 +576,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.iter().map(|x| x.num_rows()).sum::<usize>())
+            .sum();
 
         assert_eq!(8, output_partitions.len());
         assert_eq!(total_rows, 8 * 50 * 3);