File tree Expand file tree Collapse file tree 1 file changed +8
-5
lines changed Expand file tree Collapse file tree 1 file changed +8
-5
lines changed Original file line number Diff line number Diff line change @@ -2630,14 +2630,17 @@ def test_index_single_double_tuples(self):
2630
2630
# GH 20991
2631
2631
tuple_1 = tuple ([1 , 2 ])
2632
2632
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 )
2634
2635
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 )
2637
2639
tm .assert_frame_equal (result , expected )
2638
2640
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 )
2641
2644
tm .assert_frame_equal (result , expected )
2642
2645
2643
2646
def test_boolean_indexing (self ):
You can’t perform that action at this time.
0 commit comments