Skip to content

Commit

Permalink
extra tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Ben Kandel committed Oct 7, 2016
1 parent 5c0108b commit 9615a69
Showing 1 changed file with 14 additions and 6 deletions.
20 changes: 14 additions & 6 deletions pandas/tests/frame/test_combine_concat.py
Original file line number Diff line number Diff line change
Expand Up @@ -327,17 +327,25 @@ def test_join_multiindex_leftright(self):
def test_concat_named_keys(self):
# GH 14252
df = pd.DataFrame({'foo': [1, 2], 'bar': [0.1, 0.2]})
df_concatted = pd.DataFrame(
index = Index(['a', 'b'], name='baz')
concatted_named_from_keys = pd.concat([df, df], keys=index)
expected_named = 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_frame_equal(concatted_named_from_keys, df_concatted)
index_no_name = ['a', 'b']
assert_frame_equal(concatted_named_from_keys, expected_named)

index_no_name = Index(['a', 'b'], name=None)
concatted_named_from_names = pd.concat(
[df, df], keys=index_no_name, names=['baz'])
assert_frame_equal(concatted_named_from_names, df_concatted)
assert_frame_equal(concatted_named_from_names, expected_named)

concatted_unnamed = pd.concat([df, df], keys=index_no_name)
expected_unnamed = 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=[None, None]))
assert_frame_equal(concatted_unnamed, expected_unnamed)


class TestDataFrameCombineFirst(tm.TestCase, TestData):
Expand Down

0 comments on commit 9615a69

Please sign in to comment.