| 576 | reason="https://github.com/dask/dask/pull/11320#issuecomment-2293798597", |
| 577 | ) |
| 578 | def test_cumulative(): |
| 579 | index = [f"row{i:03d}" for i in range(100)] |
| 580 | df = pd.DataFrame(np.random.randn(100, 5), columns=list("abcde"), index=index) |
| 581 | |
| 582 | ddf = dd.from_pandas(df, 5) |
| 583 | |
| 584 | assert_eq(ddf.cumsum(), df.cumsum()) |
| 585 | assert_eq(ddf.cumprod(), df.cumprod()) |
| 586 | assert_eq(ddf.cummin(), df.cummin()) |
| 587 | assert_eq(ddf.cummax(), df.cummax()) |
| 588 | |
| 589 | assert_eq(ddf.cumsum(axis=1), df.cumsum(axis=1)) |
| 590 | assert_eq(ddf.cumprod(axis=1), df.cumprod(axis=1)) |
| 591 | assert_eq(ddf.cummin(axis=1), df.cummin(axis=1)) |
| 592 | assert_eq(ddf.cummax(axis=1), df.cummax(axis=1)) |
| 593 | |
| 594 | assert_eq(ddf.a.cumsum(), df.a.cumsum()) |
| 595 | assert_eq(ddf.a.cumprod(), df.a.cumprod()) |
| 596 | assert_eq(ddf.a.cummin(), df.a.cummin()) |
| 597 | assert_eq(ddf.a.cummax(), df.a.cummax()) |
| 598 | |
| 599 | assert_eq(np.cumsum(ddf), np.cumsum(df)) |
| 600 | assert_eq(np.cumprod(ddf), np.cumprod(df)) |
| 601 | assert_eq(np.cumsum(ddf, axis=1), np.cumsum(df, axis=1)) |
| 602 | assert_eq(np.cumprod(ddf, axis=1), np.cumprod(df, axis=1)) |
| 603 | assert_eq(np.cumsum(ddf.a), np.cumsum(df.a)) |
| 604 | assert_eq(np.cumprod(ddf.a), np.cumprod(df.a)) |
| 605 | |
| 606 | |
| 607 | def test_cumulative_with_nans(): |