Skip to content

Commit fdbd901

Browse files
author
Guilherme Beltramini
committed
BUG: Tests for reindex with columns and methods
1 parent 2b5724b commit fdbd901

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

pandas/tests/frame/test_axis_select_reindex.py

+29
Original file line numberDiff line numberDiff line change
@@ -297,6 +297,35 @@ def test_reindex_columns(self):
297297
newFrame = self.frame.reindex(columns=[])
298298
self.assertTrue(newFrame.empty)
299299

300+
# GH 14992, reindexing over columns ignored method
301+
df = DataFrame(data=[[11, 12, 13], [21, 22, 23], [31, 32, 33]],
302+
index=[1, 2, 4],
303+
columns=[1, 2, 4],
304+
dtype=float)
305+
expected_def = DataFrame(data=[[np.nan, 11, 12, np.nan, 13, np.nan],
306+
[np.nan, 21, 22, np.nan, 23, np.nan],
307+
[np.nan, 31, 32, np.nan, 33, np.nan]],
308+
index=[1, 2, 4],
309+
columns=range(6),
310+
dtype=float)
311+
expected_ffill = DataFrame(data=[[np.nan, 11, 12, 12, 13, 13],
312+
[np.nan, 21, 22, 22, 23, 23],
313+
[np.nan, 31, 32, 32, 33, 33]],
314+
index=[1, 2, 4],
315+
columns=range(6),
316+
dtype=float)
317+
expected_bfill = DataFrame(data=[[11, 11, 12, 13, 13, np.nan],
318+
[21, 21, 22, 23, 23, np.nan],
319+
[31, 31, 32, 33, 33, np.nan]],
320+
index=[1, 2, 4],
321+
columns=range(6),
322+
dtype=float)
323+
assert_frame_equal(df.reindex(columns=range(6)), expected_def)
324+
assert_frame_equal(df.reindex(columns=range(6), method='ffill'),
325+
expected_ffill)
326+
assert_frame_equal(df.reindex(columns=range(6), method='bfill'),
327+
expected_bfill)
328+
300329
def test_reindex_axes(self):
301330
# GH 3317, reindexing by both axes loses freq of the index
302331
df = DataFrame(np.ones((3, 3)),

0 commit comments

Comments
 (0)