Improve groupby states (#3291)

pola-rs · May 3, 2022 · c45e74c · c45e74c
1 parent 519096f
commit c45e74c
Show file tree

Hide file tree

Showing 38 changed files with 348 additions and 632 deletions.
diff --git a/polars/polars-core/src/chunked_array/logical/categorical/mod.rs b/polars/polars-core/src/chunked_array/logical/categorical/mod.rs
@@ -278,7 +278,7 @@ mod test {
         ));
 
         let groups = s.group_tuples(false, true);
-        let aggregated = s.agg_list(&groups).unwrap();
+        let aggregated = s.agg_list(&groups);
         match aggregated.get(0) {
             AnyValue::List(s) => {
                 assert!(matches!(s.dtype(), DataType::Categorical(_)));

diff --git a/polars/polars-core/src/chunked_array/object/extension/mod.rs b/polars/polars-core/src/chunked_array/object/extension/mod.rs
@@ -191,7 +191,7 @@ mod test {
         let ca = ObjectChunked::new("", values);
 
         let groups = GroupsProxy::Idx(vec![(0, vec![0, 1]), (2, vec![2]), (3, vec![3])].into());
-        let out = ca.agg_list(&groups).unwrap();
+        let out = ca.agg_list(&groups);
         assert!(matches!(out.dtype(), DataType::List(_)));
         assert_eq!(out.len(), groups.len());
     }
@@ -214,7 +214,7 @@ mod test {
         let ca = ObjectChunked::new("", values);
 
         let groups = vec![(0, vec![0, 1]), (2, vec![2]), (3, vec![3])].into();
-        let out = ca.agg_list(&GroupsProxy::Idx(groups)).unwrap();
+        let out = ca.agg_list(&GroupsProxy::Idx(groups));
         let a = out.explode().unwrap();
 
         let ca_foo = a.as_any().downcast_ref::<ObjectChunked<Foo>>().unwrap();