Skip to content

Commit 9fef291

Browse files
committed
Merge pull request #10501 from sinhrks/test_dtcategorical
TST: Fix test for datetime categorical
2 parents 4a03e93 + b8528da commit 9fef291

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

pandas/tests/test_groupby.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -3505,11 +3505,13 @@ def test_groupby_datetime_categorical(self):
35053505
desc_result = grouped.describe()
35063506

35073507
idx = cats.codes.argsort()
3508-
ord_labels = np.asarray(cats).take(idx)
3508+
ord_labels = cats.take_nd(idx)
35093509
ord_data = data.take(idx)
3510-
expected = ord_data.groupby(ord_labels, sort=False).describe()
3510+
expected = ord_data.groupby(ord_labels).describe()
35113511
expected.index.names = ['myfactor', None]
35123512
assert_frame_equal(desc_result, expected)
3513+
tm.assert_index_equal(desc_result.index, expected.index)
3514+
tm.assert_index_equal(desc_result.index.get_level_values(0), expected.index.get_level_values(0))
35133515

35143516
# GH 10460
35153517
expc = Categorical.from_codes(np.arange(4).repeat(8), levels, name='myfactor', ordered=True)

0 commit comments

Comments
 (0)