use better hashing for inner join

pola-rs · May 28, 2021 · e469bb4 · e469bb4
1 parent 0a5cd1b
commit e469bb4
Show file tree

Hide file tree

Showing 2 changed files with 229 additions and 105 deletions.
diff --git a/polars/polars-core/src/frame/groupby/mod.rs b/polars/polars-core/src/frame/groupby/mod.rs
@@ -1292,18 +1292,14 @@ mod test {
             let ca = UInt32Chunked::new_from_slice("", &slice);
             let splitted = split_ca(&ca, 4).unwrap();
 
-            let a = groupby(ca.into_iter())
+            let a = groupby(ca.into_iter()).into_iter().sorted().collect_vec();
+
+            let keys = splitted.iter().map(|ca| ca.cont_slice().unwrap()).collect();
+            let b = groupby_threaded_num(keys, 0)
                 .into_iter()
                 .sorted()
                 .collect_vec();
 
-            let keys = splitted.iter().map(|ca| ca.cont_slice().unwrap()).collect();
-            let b =
-                groupby_threaded_num(keys, 0,)
-                    .into_iter()
-                    .sorted()
-                    .collect_vec();
-
             assert_eq!(a, b);
         }
     }