From db5a9dbea89d832a70a7a57ad8049afc7c603ca4 Mon Sep 17 00:00:00 2001 From: ritchie Date: Tue, 10 Sep 2024 13:27:10 +0200 Subject: [PATCH] wip [skip ci] --- py-polars/tests/unit/datatypes/test_list.py | 2 +- .../tests/unit/operations/aggregation/test_aggregations.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/py-polars/tests/unit/datatypes/test_list.py b/py-polars/tests/unit/datatypes/test_list.py index 4607cfa894260..b01f4fd77e474 100644 --- a/py-polars/tests/unit/datatypes/test_list.py +++ b/py-polars/tests/unit/datatypes/test_list.py @@ -297,7 +297,7 @@ def test_flat_aggregation_to_list_conversion_6918() -> None: assert df.group_by("a", maintain_order=True).agg( pl.concat_list([pl.col("b").list.get(i).mean().implode() for i in range(2)]) - ).to_dict(as_series=False) == {"a": [1, 2], "b": [[[0.0, 1.0]], [[3.0, 4.0]]]} + ).to_dict(as_series=False) == {"a": [1, 2], "b": [[0.0, 1.0], [3.0, 4.0]]} def test_list_count_matches() -> None: diff --git a/py-polars/tests/unit/operations/aggregation/test_aggregations.py b/py-polars/tests/unit/operations/aggregation/test_aggregations.py index 8667b419d59e5..016b17a75badb 100644 --- a/py-polars/tests/unit/operations/aggregation/test_aggregations.py +++ b/py-polars/tests/unit/operations/aggregation/test_aggregations.py @@ -258,7 +258,7 @@ def test_err_on_implode_and_agg() -> None: pl.col("type").implode().list.head().alias("foo") ).to_dict(as_series=False) == { "type": ["water", "fire", "earth"], - "foo": [[["water", "water"]], [["fire"]], [["earth"]]], + "foo": [["water", "water"], ["fire"], ["earth"]], } # but not during a window function as the groups cannot be mapped back