File tree Expand file tree Collapse file tree 1 file changed +8
-5
lines changed
pandas/tests/frame/indexing Expand file tree Collapse file tree 1 file changed +8
-5
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments