comments

Author: Ben Kandel

pandas/tests/frame/test_combine_concat.py

@@ -12,13 +12,10 @@
 from pandas import DataFrame, Index, Series, Timestamp
 from pandas.compat import lrange
 
-from pandas.core.base import FrozenList
-
 from pandas.tests.frame.common import TestData
 
 import pandas.util.testing as tm
 from pandas.util.testing import (assertRaisesRegexp,
-                                 assert_equal,
                                  assert_frame_equal,
                                  assert_series_equal)
 
@@ -329,14 +326,18 @@ def test_join_multiindex_leftright(self):
 
     def test_concat_named_keys(self):
         # GH 14252
-        df = DataFrame({'foo': [1, 2, 3, 4],
-                        'bar': [0.1, 0.2, 0.3, 0.4]})
+        df = pd.DataFrame({'foo': [1, 2], 'bar': [0.1, 0.2]})
+        df_concatted = pd.DataFrame(
+            {'foo': [1, 2, 1, 2], 'bar': [0.1, 0.2, 0.1, 0.2]},
+            index=pd.MultiIndex.from_product((['a', 'b'], [0, 1]),
+                                             names=['baz', None]))
         index = Index(['a', 'b'], name='baz')
         concatted_named_from_keys = pd.concat([df, df], keys=index)
-        assert_equal(concatted_named_from_keys.index.names, FrozenList(['baz', None]))
+        assert_frame_equal(concatted_named_from_keys, df_concatted)
         index_no_name = ['a', 'b']
-        concatted_named_from_names = pd.concat([df, df], keys=index_no_name, names=['baz'])
-        assert_equal(concatted_named_from_names.index.names, FrozenList(['baz', None]))
+        concatted_named_from_names = pd.concat(
+            [df, df], keys=index_no_name, names=['baz'])
+        assert_frame_equal(concatted_named_from_names, df_concatted)
 
 
 class TestDataFrameCombineFirst(tm.TestCase, TestData):