Skip to content

Commit 10a5305

Browse files
TST: test that re-using a list of groupers will not throw ValueError (#62374)
1 parent 4ffdfe9 commit 10a5305

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

pandas/tests/groupby/test_grouping.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -875,6 +875,22 @@ def test_groupby_tuple_keys_handle_multiindex(self):
875875
expected = expected[result.columns]
876876
tm.assert_frame_equal(result.reset_index(drop=True), expected)
877877

878+
def test_groupby_grouper_immutable_list_item(self):
879+
# GH 26564 - prevent 'ValueError: all keys need to be the same shape'
880+
# when reusing a list of groupers
881+
df1 = DataFrame([["05/29/2019"], ["05/28/2019"]], columns=["date"]).assign(
882+
date=lambda df: pd.to_datetime(df["date"])
883+
)
884+
df2 = DataFrame(columns=["date"]).assign(
885+
date=lambda df: pd.to_datetime(df["date"])
886+
)
887+
888+
groupers = [Grouper(key="date", freq="1D")]
889+
890+
df1.groupby(groupers).head()
891+
# no error
892+
df2.groupby(groupers).head()
893+
878894

879895
# get_group
880896
# --------------------------------

0 commit comments

Comments
 (0)