Skip to content

Commit db7866f

Browse files
committed
construct expected results as DataFrame instead of FrozenList
1 parent 7f82be9 commit db7866f

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

pandas/tests/tools/test_concat.py

+12-8
Original file line numberDiff line numberDiff line change
@@ -1050,21 +1050,25 @@ def test_concat_multiindex_with_tz(self):
10501050

10511051
def test_concat_multiindex_with_none_in_index_names(self):
10521052
# GH 15787
1053-
from pandas.indexes.frozen import FrozenList
1054-
10551053
index = pd.MultiIndex.from_product([[1], range(5)],
10561054
names=['level1', None])
10571055
df = pd.DataFrame({'col': range(5)}, index=index)
10581056

10591057
result = concat([df, df], keys=[1, 2], names=['level2'])
1060-
result = result.index.names
1061-
expected = FrozenList(['level2', 'level1', None])
1062-
self.assertEqual(result, expected)
1058+
index = pd.MultiIndex.from_product([[1, 2], [1], range(5)],
1059+
names=['level2', 'level1', None])
1060+
expected = pd.DataFrame({'col': list(range(5)) * 2}, index=index)
1061+
assert_frame_equal(result, expected)
10631062

10641063
result = concat([df, df[:2]], keys=[1, 2], names=['level2'])
1065-
result = result.index.names
1066-
expected = FrozenList(['level2', 'level1', None])
1067-
self.assertEqual(result, expected)
1064+
level2 = [1] * 5 + [2] * 2
1065+
level1 = [1] * 7
1066+
no_name = list(range(5)) + list(range(2))
1067+
tuples = list(zip(level2, level1, no_name))
1068+
index = pd.MultiIndex.from_tuples(tuples,
1069+
names=['level2', 'level1', None])
1070+
expected = pd.DataFrame({'col': no_name}, index=index)
1071+
assert_frame_equal(result, expected)
10681072

10691073
def test_concat_keys_and_levels(self):
10701074
df = DataFrame(np.random.randn(1, 3))

0 commit comments

Comments
 (0)