Skip to content

Commit 7efc4b8

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

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/test_sorting.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,21 @@ def test_stable_descending_multicolumn_sort(self):
227227
kind='mergesort')
228228
assert_frame_equal(sorted_df, expected)
229229

230+
@pytest.mark.parametrize('seed', list(range(100)))
231+
def test_sort_multi_index(self, seed):
232+
# GH 25775
233+
pd.np.random.seed(seed)
234+
df1 = (pd.DataFrame(pd.np.random.rand(10, 5),
235+
columns=['a', 'b', 'c', 'd', 'e'])
236+
.assign(b=lambda df: (df.b * 10).astype(int)))
237+
df2 = df1.set_index(['a', 'b', 'c']).sort_index(
238+
axis=0, level=['b', 'a'])
239+
expected = pd.DataFrame(df2.index.to_frame()[['a', 'b']].values,
240+
columns=['a', 'b']).assign(
241+
b=lambda df: (df.b).astype(int))
242+
result = df1.sort_values(by=['b', 'a']).reset_index()[['a', 'b']]
243+
tm.assert_frame_equal(result, expected)
244+
230245
def test_stable_categorial(self):
231246
# GH 16793
232247
df = DataFrame({

0 commit comments

Comments
 (0)