Skip to content

Commit 43a3f0e

Browse files
authored
TST: added test for fixed bug gh#50372 (#54701)
* added test for bug gh#50372 * cleaned code to fit pre-commit * moved test to better fitting location
1 parent 00f79a3 commit 43a3f0e

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

pandas/tests/indexing/multiindex/test_getitem.py

+17
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,23 @@ def test_frame_getitem_simple_key_error(
145145
indexer(df)
146146

147147

148+
def test_tuple_string_column_names():
149+
# GH#50372
150+
mi = MultiIndex.from_tuples([("a", "aa"), ("a", "ab"), ("b", "ba"), ("b", "bb")])
151+
df = DataFrame([range(0, 4), range(1, 5), range(2, 6)], columns=mi)
152+
df["single_index"] = 0
153+
154+
df_flat = df.copy()
155+
df_flat.columns = df_flat.columns.to_flat_index()
156+
df_flat["new_single_index"] = 0
157+
158+
result = df_flat[[("a", "aa"), "new_single_index"]]
159+
expected = DataFrame(
160+
[[0, 0], [1, 0], [2, 0]], columns=Index([("a", "aa"), "new_single_index"])
161+
)
162+
tm.assert_frame_equal(result, expected)
163+
164+
148165
def test_frame_getitem_multicolumn_empty_level():
149166
df = DataFrame({"a": ["1", "2", "3"], "b": ["2", "3", "4"]})
150167
df.columns = [

0 commit comments

Comments
 (0)