Skip to content

Commit 9615a69

Browse files
author
Ben Kandel
committed
extra tests
1 parent 5c0108b commit 9615a69

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

pandas/tests/frame/test_combine_concat.py

+14-6
Original file line numberDiff line numberDiff line change
@@ -327,17 +327,25 @@ def test_join_multiindex_leftright(self):
327327
def test_concat_named_keys(self):
328328
# GH 14252
329329
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(
331333
{'foo': [1, 2, 1, 2], 'bar': [0.1, 0.2, 0.1, 0.2]},
332334
index=pd.MultiIndex.from_product((['a', 'b'], [0, 1]),
333335
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)
338339
concatted_named_from_names = pd.concat(
339340
[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)
341349

342350

343351
class TestDataFrameCombineFirst(tm.TestCase, TestData):

0 commit comments

Comments
 (0)