Skip to content

Commit 155ffc0

Browse files
committed
TST: test that re-using a list of groupers will not throw ValueError
1 parent b2a6d74 commit 155ffc0

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

pandas/tests/groupby/test_grouping.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -875,6 +875,25 @@ 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' when re-using a list of groupers
880+
df1 = pd.DataFrame(
881+
[
882+
["05/29/2019"],
883+
["05/28/2019"]
884+
],
885+
columns=["date"]).assign(date=lambda df: pd.to_datetime(df["date"])
886+
)
887+
df2 = pd.DataFrame(
888+
columns=["date"]).assign(date=lambda df: pd.to_datetime(df["date"])
889+
)
890+
891+
groupers = [pd.Grouper(key="date", freq="1D")]
892+
893+
df1.groupby(groupers).head()
894+
# no error
895+
df2.groupby(groupers).head()
896+
878897

879898
# get_group
880899
# --------------------------------

0 commit comments

Comments
 (0)