Skip to content

Commit 1763b74

Browse files
author
Matias Heikkilä
committed
TST: Test sorting levels not aligned with index (#25775)
1 parent 6d2398a commit 1763b74

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

pandas/tests/frame/test_sorting.py

+13
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from io import StringIO
12
import random
23

34
import numpy as np
@@ -227,6 +228,18 @@ def test_stable_descending_multicolumn_sort(self):
227228
kind='mergesort')
228229
assert_frame_equal(sorted_df, expected)
229230

231+
def test_sort_multi_index(self):
232+
# GH 25775
233+
df = DataFrame({'a': [3, 1, 2], 'b': [0, 0, 0],
234+
'c': [0, 1, 2], 'd': list('abc')})
235+
result = df.set_index(list('abc')).sort_index(level=list('ba'))
236+
237+
expected = DataFrame({'a': [1, 2, 3], 'b': [0, 0, 0],
238+
'c': [1, 2, 0], 'd': list('bca')})
239+
expected = expected.set_index(list('abc'))
240+
241+
tm.assert_frame_equal(result, expected)
242+
230243
def test_stable_categorial(self):
231244
# GH 16793
232245
df = DataFrame({

0 commit comments

Comments
 (0)