55
66
77class TestMultiIndexInsertLevel :
8-
98 def setup_method (self ):
109 self .simple_idx = pd .MultiIndex .from_tuples (
1110 [("A" , 1 ), ("B" , 2 ), ("C" , 3 )], names = ["level1" , "level2" ]
@@ -16,14 +15,14 @@ def test_insert_level_basic(self):
1615 result = self .simple_idx .insert_level (0 , "new_value" )
1716 expected = pd .MultiIndex .from_tuples (
1817 [("new_value" , "A" , 1 ), ("new_value" , "B" , 2 ), ("new_value" , "C" , 3 )],
19- names = [None , "level1" , "level2" ]
18+ names = [None , "level1" , "level2" ],
2019 )
2120 tm .assert_index_equal (result , expected )
2221
2322 result = self .simple_idx .insert_level (1 , "middle" )
2423 expected = pd .MultiIndex .from_tuples (
2524 [("A" , "middle" , 1 ), ("B" , "middle" , 2 ), ("C" , "middle" , 3 )],
26- names = ["level1" , None , "level2" ]
25+ names = ["level1" , None , "level2" ],
2726 )
2827 tm .assert_index_equal (result , expected )
2928
@@ -32,7 +31,7 @@ def test_insert_level_with_different_values(self):
3231 result = self .simple_idx .insert_level (1 , new_values )
3332 expected = pd .MultiIndex .from_tuples (
3433 [("A" , "X" , 1 ), ("B" , "Y" , 2 ), ("C" , "Z" , 3 )],
35- names = ["level1" , None , "level2" ]
34+ names = ["level1" , None , "level2" ],
3635 )
3736 tm .assert_index_equal (result , expected )
3837
@@ -78,8 +77,7 @@ def test_insert_level_preserves_original(self):
7877
7978 def test_debug_names ():
8079 idx = pd .MultiIndex .from_tuples (
81- [("A" , 1 ), ("B" , 2 ), ("C" , 3 )],
82- names = ["level1" , "level2" ]
80+ [("A" , 1 ), ("B" , 2 ), ("C" , 3 )], names = ["level1" , "level2" ]
8381 )
8482 print ("Original names:" , idx .names )
8583
@@ -88,6 +86,6 @@ def test_debug_names():
8886
8987 expected = pd .MultiIndex .from_tuples (
9088 [("new_value" , "A" , 1 ), ("new_value" , "B" , 2 ), ("new_value" , "C" , 3 )],
91- names = [None , "level1" , "level2" ]
89+ names = [None , "level1" , "level2" ],
9290 )
9391 print ("Expected names:" , expected .names )
0 commit comments