Skip to content

Commit 3f08b07

Browse files
Adding concat tests for axis 0, 1, and 'index'
1 parent cf3f998 commit 3f08b07

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

pandas/tests/frame/test_combine_concat.py

+15
Original file line numberDiff line numberDiff line change
@@ -356,11 +356,26 @@ def test_concat_axis_parameter(self):
356356
concatted_row = pd.concat([df1, df2], axis='rows')
357357
assert_frame_equal(concatted_row, expected_row)
358358

359+
expected_index = pd.DataFrame(
360+
{'A': [0.1, 0.2, 0.3, 0.4]}, index=[0, 1, 0, 1])
361+
concatted_index = pd.concat([df1, df2], axis='index')
362+
assert_frame_equal(concatted_index, expected_index)
363+
364+
expected_0 = pd.DataFrame(
365+
{'A': [0.1, 0.2, 0.3, 0.4]}, index=[0, 1, 0, 1])
366+
concatted_0 = pd.concat([df1, df2], axis=0)
367+
assert_frame_equal(concatted_0, expected_0)
368+
359369
expected_columns = pd.DataFrame(
360370
[[0.1, 0.3], [0.2, 0.4]], index=[0, 1], columns=['A', 'A'])
361371
concatted_columns = pd.concat([df1, df2], axis='columns')
362372
assert_frame_equal(concatted_columns, expected_columns)
363373

374+
expected_1 = pd.DataFrame(
375+
[[0.1, 0.3], [0.2, 0.4]], index=[0, 1], columns=['A', 'A'])
376+
concatted_1 = pd.concat([df1, df2], axis=1)
377+
assert_frame_equal(concatted_1, expected_1)
378+
364379
series1 = pd.Series([0.1, 0.2])
365380
series2 = pd.Series([0.3, 0.4])
366381

0 commit comments

Comments
 (0)