Commit 10a5305

authored

TST: test that re-using a list of groupers will not throw ValueError (#62374)

1 parent 4ffdfe9 commit 10a5305Copy full SHA for 10a5305

File tree

+16

-0

lines changed

+16

-0

lines changed

Lines changed: 16 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -875,6 +875,22 @@ def test_groupby_tuple_keys_handle_multiindex(self):`
`875`	`875`	`expected = expected[result.columns]`
`876`	`876`	`tm.assert_frame_equal(result.reset_index(drop=True), expected)`
`877`	`877`
	`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`	`+`
`878`	`894`
`879`	`895`	`# get_group`
`880`	`896`	`# --------------------------------`

Comments

(0)