Skip to content

Commit 359231c

Browse files
committed
Fix categorical test
1 parent 5f8477f commit 359231c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

pandas/tests/groupby/aggregate/test_aggregate.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -463,9 +463,12 @@ def test_agg_cython_category_not_implemented_fallback():
463463
df = pd.DataFrame({"col_num": [1, 1, 2, 3]})
464464
df["col_cat"] = df["col_num"].astype("category")
465465

466-
result = df.groupby("col_num").col_cat.first()
466+
result = df.groupby("col_num")["col_cat"].first()
467467
expected = pd.Series(
468-
[1, 2, 3], index=pd.Index([1, 2, 3], name="col_num"), name="col_cat"
468+
[1, 2, 3],
469+
dtype="category",
470+
index=pd.Index([1, 2, 3], name="col_num"),
471+
name="col_cat",
469472
)
470473
tm.assert_series_equal(result, expected)
471474

0 commit comments

Comments
 (0)