| 904 | ], |
| 905 | ) |
| 906 | def test_as_index_false(self, by, expected_data, unit): |
| 907 | # GH 39433 |
| 908 | data = [ |
| 909 | ["A", "2018-01-01", 100.0], |
| 910 | ["A", "2018-01-02", 200.0], |
| 911 | ["B", "2018-01-01", 150.0], |
| 912 | ["B", "2018-01-02", 250.0], |
| 913 | ] |
| 914 | df = DataFrame(data, columns=["id", "date", "num"]) |
| 915 | df["date"] = df["date"].astype(f"M8[{unit}]") |
| 916 | df = df.set_index(["date"]) |
| 917 | |
| 918 | gp_by = [getattr(df, attr) for attr in by] |
| 919 | result = ( |
| 920 | df.groupby(gp_by, as_index=False).rolling(window=2, min_periods=1).mean() |
| 921 | ) |
| 922 | |
| 923 | expected = {"id": ["A", "A", "B", "B"]} |
| 924 | expected.update(expected_data) |
| 925 | expected = DataFrame( |
| 926 | expected, |
| 927 | index=df.index, |
| 928 | ) |
| 929 | if "date" in expected_data: |
| 930 | expected["date"] = expected["date"].astype(f"M8[{unit}]") |
| 931 | tm.assert_frame_equal(result, expected) |
| 932 | |
| 933 | def test_nan_and_zero_endpoints(self, any_int_numpy_dtype): |
| 934 | # https://github.com/twosigma/pandas/issues/53 |