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):
2628
2628
# GH 20991
2629
2629
tuple_1 = tuple ([1 , 2 ])
2630
2630
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 )
2632
2633
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 )
2635
2637
tm .assert_frame_equal (result , expected )
2636
2638
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 )
2639
2642
tm .assert_frame_equal (result , expected )
2640
2643
2641
2644
def test_boolean_indexing (self ):
You can’t perform that action at this time.
0 commit comments