BUG: fix test in rolling groupby (pandas-dev#44821)

maurosilber · maurosilber · commit 7f8005defc6b · 2021-12-19T16:42:08.000-03:00
Fixes test due to changes in GroupBy.__iter__ (see pandas-dev#44821). As the column `c` wasn't selected on the manual computation, it failed when trying to set it as an index.
diff --git a/pandas/tests/window/test_base_indexer.py b/pandas/tests/window/test_base_indexer.py
@@ -437,7 +437,7 @@ def test_rolling_groupby_with_fixed_forward_many(group_keys, window_size):
     result = df.groupby("a")["b"].rolling(window=indexer, min_periods=1).sum()
     result.index.names = ["a", "c"]
 
-    groups = df.groupby("a")[["a", "b"]]
+    groups = df.groupby("a")[["a", "b", "c"]]
     manual = concat(
         [
             g.assign(

Original file line number	Diff line number	Diff line change
`@@ -437,7 +437,7 @@ def test_rolling_groupby_with_fixed_forward_many(group_keys, window_size):`
`437`	`437`	`result = df.groupby("a")["b"].rolling(window=indexer, min_periods=1).sum()`
`438`	`438`	`result.index.names = ["a", "c"]`
`439`	`439`
`440`		`- groups = df.groupby("a")[["a", "b"]]`
	`440`	`+ groups = df.groupby("a")[["a", "b", "c"]]`
`441`	`441`	`manual = concat(`
`442`	`442`	`[`
`443`	`443`	`g.assign(`