Skip to content

Commit 3984ce1

Browse files
authored
test: Test for group_by(...).having(...) (pola-rs#25430)
1 parent 13c0d8e commit 3984ce1

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

py-polars/tests/unit/operations/test_group_by.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1474,17 +1474,22 @@ def test_group_by_shift_filter_23910(maintain_order: bool) -> None:
14741474
)
14751475

14761476

1477-
def group_by_having() -> None:
1477+
@pytest.mark.parametrize("maintain_order", [False, True])
1478+
def test_group_by_having(maintain_order: bool) -> None:
14781479
df = pl.DataFrame(
14791480
{
14801481
"grp": ["A", "A", "B", "B", "C", "C"],
14811482
"value": [10, 15, 5, 15, 5, 10],
14821483
}
14831484
)
14841485

1485-
result = df.group_by("group").having(pl.col("value").mean() >= 10).agg()
1486+
result = (
1487+
df.group_by("grp", maintain_order=maintain_order)
1488+
.having(pl.col("value").mean() >= 10)
1489+
.agg()
1490+
)
14861491
expected = pl.DataFrame({"grp": ["A", "B"]})
1487-
assert_frame_equal(result, expected, check_row_order=False)
1492+
assert_frame_equal(result, expected, check_row_order=maintain_order)
14881493

14891494

14901495
def test_group_by_tuple_typing_24112() -> None:

0 commit comments

Comments
 (0)