Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
pola-rs
GitHub Repository: pola-rs/polars
Path: blob/main/py-polars/tests/unit/expr/test_expr_apply_eval.py
6939 views
1
import polars as pl
2
3
4
def test_expression_15183() -> None:
5
assert (
6
pl.DataFrame(
7
{"a": [1, 2, 3, 4, 5, 2, 3, 5, 1], "b": [1, 2, 3, 1, 2, 3, 1, 2, 3]}
8
)
9
.group_by("a")
10
.agg(pl.col.b.unique().sort().str.join("-").str.split("-"))
11
.sort("a")
12
).to_dict(as_series=False) == {
13
"a": [1, 2, 3, 4, 5],
14
"b": [["1", "3"], ["2", "3"], ["1", "3"], ["1"], ["2"]],
15
}
16
17