Skip to content

Commit 9652128

Browse files
committed
remove .set_index() and .index[]
1 parent acafc9e commit 9652128

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

pandas/tests/frame/indexing/test_indexing.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2628,14 +2628,17 @@ def test_index_single_double_tuples(self):
26282628
# GH 20991
26292629
tuple_1 = tuple([1, 2])
26302630
tuple_2 = tuple([1])
2631-
df = DataFrame([[tuple_1], [tuple_2]], columns=["A"]).set_index("A")
2631+
idx = pd.Index([tuple_1, tuple_2], name="A", tupleize_cols=False)
2632+
df = DataFrame(index=idx)
26322633

2633-
result = df.loc[[df.index[0]]]
2634-
expected = DataFrame([[tuple_1]], columns=["A"]).set_index("A")
2634+
result = df.loc[[tuple_1]]
2635+
idx_1 = pd.Index([tuple_1], name="A", tupleize_cols=False)
2636+
expected = DataFrame(index=idx_1)
26352637
tm.assert_frame_equal(result, expected)
26362638

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

26412644
def test_boolean_indexing(self):

0 commit comments

Comments
 (0)