Skip to content

Commit 76f58fc

Browse files
committed
add test for indexing with single/double tuple
1 parent 0de9955 commit 76f58fc

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

pandas/tests/frame/test_indexing.py

+14
Original file line numberDiff line numberDiff line change
@@ -2626,6 +2626,20 @@ def test_index_namedtuple(self):
26262626
result = df.loc[IndexType("foo", "bar")]["A"]
26272627
assert result == 1
26282628

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

0 commit comments

Comments
 (0)