()
| 470 | |
| 471 | |
| 472 | def test_groupby_index_array(): |
| 473 | df = _compat.makeTimeDataFrame() |
| 474 | ddf = dd.from_pandas(df, npartitions=2) |
| 475 | |
| 476 | # first select column, then group |
| 477 | assert_eq( |
| 478 | df.A.groupby(df.index.month).nunique(), |
| 479 | ddf.A.groupby(ddf.index.month).nunique(), |
| 480 | check_names=False, |
| 481 | ) |
| 482 | |
| 483 | # first group, then select column |
| 484 | assert_eq( |
| 485 | df.groupby(df.index.month).A.nunique(), |
| 486 | ddf.groupby(ddf.index.month).A.nunique(), |
| 487 | check_names=False, |
| 488 | ) |
| 489 | |
| 490 | |
| 491 | def test_groupby_set_index(): |