Skip to content

Commit 4cfab61

Browse files
committed
add test for indexing with single/double tuple
1 parent 6b3ba98 commit 4cfab61

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

pandas/tests/frame/indexing/test_indexing.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2624,6 +2624,20 @@ def test_index_namedtuple(self):
26242624
result = df.loc[IndexType("foo", "bar")]["A"]
26252625
assert result == 1
26262626

2627+
def test_index_single_double_tuples(self):
2628+
# GH 20991
2629+
tuple_1 = tuple([1, 2])
2630+
tuple_2 = tuple([1])
2631+
df = pd.DataFrame([[tuple_1], [tuple_2]], columns=["A"]).set_index("A")
2632+
2633+
result = df.loc[[df.index[0]]]
2634+
expected = pd.DataFrame([[tuple_1]], columns=["A"]).set_index("A")
2635+
tm.assert_frame_equal(result, expected)
2636+
2637+
result = df.loc[[df.index[1]]]
2638+
expected = pd.DataFrame([[tuple_2]], columns=["A"]).set_index("A")
2639+
tm.assert_frame_equal(result, expected)
2640+
26272641
def test_boolean_indexing(self):
26282642
idx = list(range(3))
26292643
cols = ["A", "B", "C"]

0 commit comments

Comments
 (0)