| 302 | |
| 303 | |
| 304 | def test_center_reindex_frame(raw): |
| 305 | # shifter index |
| 306 | frame = DataFrame(range(100), index=date_range("2020-01-01", freq="D", periods=100)) |
| 307 | s = [f"x{x:d}" for x in range(12)] |
| 308 | minp = 10 |
| 309 | |
| 310 | frame_xp = ( |
| 311 | frame.reindex(list(frame.index) + s) |
| 312 | .rolling(window=25, min_periods=minp) |
| 313 | .apply(f, raw=raw) |
| 314 | .shift(-12) |
| 315 | .reindex(frame.index) |
| 316 | ) |
| 317 | frame_rs = frame.rolling(window=25, min_periods=minp, center=True).apply(f, raw=raw) |
| 318 | tm.assert_frame_equal(frame_xp, frame_rs) |