From 9819be48063a98d3df23696bbb21b09183e73cde Mon Sep 17 00:00:00 2001 From: Dale Jung Date: Tue, 17 Sep 2013 23:00:06 -0400 Subject: [PATCH] TST: test that DataFrame.shift(axis='columns') works --- pandas/tests/test_frame.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/pandas/tests/test_frame.py b/pandas/tests/test_frame.py index 4758670517df0..e16c8f110323c 100644 --- a/pandas/tests/test_frame.py +++ b/pandas/tests/test_frame.py @@ -9087,7 +9087,6 @@ def test_shift(self): assertRaisesRegexp(ValueError, 'does not match PeriodIndex freq', ps.shift, freq='D') - # shift other axis # GH 6371 df = DataFrame(np.random.rand(10,5)) @@ -9095,6 +9094,15 @@ def test_shift(self): result = df.shift(1,axis=1) assert_frame_equal(result,expected) + def test_shift_columns(self): + df = tm.makeTimeDataFrame() + + shifted = df.shift(1, axis='columns') + self.assert_(shifted['A'].isnull().all()) + + self.assert_(shifted.columns.equals(df.columns)) + tm.assert_series_equal(shifted['B'], df['A']) + def test_shift_bool(self): df = DataFrame({'high': [True, False], 'low': [False, False]})