From b8528da859e10e9215bd2eb10efdc9ed7ffadb34 Mon Sep 17 00:00:00 2001 From: Sinhrks Date: Fri, 3 Jul 2015 12:39:58 +0900 Subject: [PATCH] TST: Fix test for datetime categorical --- pandas/tests/test_groupby.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pandas/tests/test_groupby.py b/pandas/tests/test_groupby.py index a327233e09003..e2a447207db82 100644 --- a/pandas/tests/test_groupby.py +++ b/pandas/tests/test_groupby.py @@ -3505,11 +3505,13 @@ def test_groupby_datetime_categorical(self): desc_result = grouped.describe() idx = cats.codes.argsort() - ord_labels = np.asarray(cats).take(idx) + ord_labels = cats.take_nd(idx) ord_data = data.take(idx) - expected = ord_data.groupby(ord_labels, sort=False).describe() + expected = ord_data.groupby(ord_labels).describe() expected.index.names = ['myfactor', None] assert_frame_equal(desc_result, expected) + tm.assert_index_equal(desc_result.index, expected.index) + tm.assert_index_equal(desc_result.index.get_level_values(0), expected.index.get_level_values(0)) # GH 10460 expc = Categorical.from_codes(np.arange(4).repeat(8), levels, name='myfactor', ordered=True)