Skip to content

Commit 1e30ac7

Browse files
author
Kristian Holsheimer
committed
Added test coverage for multi-level (tuple) lookups.
1 parent 2f05ba4 commit 1e30ac7

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/tests/frame/test_analytics.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2260,3 +2260,14 @@ def test_series_nat_conversion(self):
22602260
df.rank()
22612261
result = df
22622262
tm.assert_frame_equal(result, expected)
2263+
2264+
def test_multiindex_column_lookup(self):
2265+
df = pd.DataFrame(
2266+
columns=pd.MultiIndex.from_product([['x'], ['a', 'b']]),
2267+
data=[[0.33, 0.13], [0.86, 0.25], [0.25, 0.70], [0.85, 0.91]])
2268+
pd.util.testing.assert_frame_equal(
2269+
df.nsmallest(3, ('x', 'a')),
2270+
df.iloc[[2, 0, 3]])
2271+
pd.util.testing.assert_frame_equal(
2272+
df.nlargest(3, ('x', 'b')),
2273+
df.iloc[[3, 2, 1]])

0 commit comments

Comments
 (0)