Skip to content

Commit 12a8e20

Browse files
committed
remove .set_index() and .index[]
1 parent 039c18a commit 12a8e20

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

pandas/tests/frame/test_indexing.py

+8-5
Original file line numberDiff line numberDiff line change
@@ -2630,14 +2630,17 @@ def test_index_single_double_tuples(self):
26302630
# GH 20991
26312631
tuple_1 = tuple([1, 2])
26322632
tuple_2 = tuple([1])
2633-
df = DataFrame([[tuple_1], [tuple_2]], columns=["A"]).set_index("A")
2633+
idx = pd.Index([tuple_1, tuple_2], name="A", tupleize_cols=False)
2634+
df = DataFrame(index=idx)
26342635

2635-
result = df.loc[[df.index[0]]]
2636-
expected = DataFrame([[tuple_1]], columns=["A"]).set_index("A")
2636+
result = df.loc[[tuple_1]]
2637+
idx_1 = pd.Index([tuple_1], name="A", tupleize_cols=False)
2638+
expected = DataFrame(index=idx_1)
26372639
tm.assert_frame_equal(result, expected)
26382640

2639-
result = df.loc[[df.index[1]]]
2640-
expected = DataFrame([[tuple_2]], columns=["A"]).set_index("A")
2641+
result = df.loc[[tuple_2]]
2642+
idx_2 = pd.Index([tuple_2], name="A", tupleize_cols=False)
2643+
expected = DataFrame(index=idx_2)
26412644
tm.assert_frame_equal(result, expected)
26422645

26432646
def test_boolean_indexing(self):

0 commit comments

Comments
 (0)