Skip to content

Commit 09cc554

Browse files
author
y-p
committed
TST: mom._center_window should validate axis value against data dims
1 parent 18d3893 commit 09cc554

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

Diff for: pandas/stats/tests/test_moments.py

+12
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,18 @@ def setUp(self):
3838
self.frame = DataFrame(randn(N, K), index=self.rng,
3939
columns=np.arange(K))
4040

41+
def test_centered_axis_validation(self):
42+
# ok
43+
mom.rolling_mean(Series(np.ones(10)),3,center=True ,axis=0)
44+
# bad axis
45+
self.assertRaises(ValueError, mom.rolling_mean,Series(np.ones(10)),3,center=True ,axis=1)
46+
47+
# ok ok
48+
mom.rolling_mean(DataFrame(np.ones((10,10))),3,center=True ,axis=0)
49+
mom.rolling_mean(DataFrame(np.ones((10,10))),3,center=True ,axis=1)
50+
# bad axis
51+
self.assertRaises(ValueError, mom.rolling_mean,DataFrame(np.ones((10,10))),3,center=True ,axis=2)
52+
4153
def test_rolling_sum(self):
4254
self._check_moment_func(mom.rolling_sum, np.sum)
4355

0 commit comments

Comments
 (0)