From e40c5a97cca1047838e0426986b8cc973dd7b53d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Matias=20Heikkil=C3=A4?= Date: Sun, 26 May 2019 15:29:49 +0300 Subject: [PATCH] TST: Test sorting levels not aligned with index (#25775) --- pandas/tests/frame/test_sorting.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/pandas/tests/frame/test_sorting.py b/pandas/tests/frame/test_sorting.py index 246ba943a4509..96aeb608ba3b8 100644 --- a/pandas/tests/frame/test_sorting.py +++ b/pandas/tests/frame/test_sorting.py @@ -227,6 +227,18 @@ def test_stable_descending_multicolumn_sort(self): kind='mergesort') assert_frame_equal(sorted_df, expected) + def test_sort_multi_index(self): + # GH 25775, testing that sorting by index works with a multi-index. + df = DataFrame({'a': [3, 1, 2], 'b': [0, 0, 0], + 'c': [0, 1, 2], 'd': list('abc')}) + result = df.set_index(list('abc')).sort_index(level=list('ba')) + + expected = DataFrame({'a': [1, 2, 3], 'b': [0, 0, 0], + 'c': [1, 2, 0], 'd': list('bca')}) + expected = expected.set_index(list('abc')) + + tm.assert_frame_equal(result, expected) + def test_stable_categorial(self): # GH 16793 df = DataFrame({