@@ -327,17 +327,25 @@ def test_join_multiindex_leftright(self):
327
327
def test_concat_named_keys (self ):
328
328
# GH 14252
329
329
df = pd .DataFrame ({'foo' : [1 , 2 ], 'bar' : [0.1 , 0.2 ]})
330
- df_concatted = pd .DataFrame (
330
+ index = Index (['a' , 'b' ], name = 'baz' )
331
+ concatted_named_from_keys = pd .concat ([df , df ], keys = index )
332
+ expected_named = pd .DataFrame (
331
333
{'foo' : [1 , 2 , 1 , 2 ], 'bar' : [0.1 , 0.2 , 0.1 , 0.2 ]},
332
334
index = pd .MultiIndex .from_product ((['a' , 'b' ], [0 , 1 ]),
333
335
names = ['baz' , None ]))
334
- index = Index (['a' , 'b' ], name = 'baz' )
335
- concatted_named_from_keys = pd .concat ([df , df ], keys = index )
336
- assert_frame_equal (concatted_named_from_keys , df_concatted )
337
- index_no_name = ['a' , 'b' ]
336
+ assert_frame_equal (concatted_named_from_keys , expected_named )
337
+
338
+ index_no_name = Index (['a' , 'b' ], name = None )
338
339
concatted_named_from_names = pd .concat (
339
340
[df , df ], keys = index_no_name , names = ['baz' ])
340
- assert_frame_equal (concatted_named_from_names , df_concatted )
341
+ assert_frame_equal (concatted_named_from_names , expected_named )
342
+
343
+ concatted_unnamed = pd .concat ([df , df ], keys = index_no_name )
344
+ expected_unnamed = pd .DataFrame (
345
+ {'foo' : [1 , 2 , 1 , 2 ], 'bar' : [0.1 , 0.2 , 0.1 , 0.2 ]},
346
+ index = pd .MultiIndex .from_product ((['a' , 'b' ], [0 , 1 ]),
347
+ names = [None , None ]))
348
+ assert_frame_equal (concatted_unnamed , expected_unnamed )
341
349
342
350
343
351
class TestDataFrameCombineFirst (tm .TestCase , TestData ):
0 commit comments