def test_dataframe(self):
x = DataFrame(randn(2, 3))
for axis, ddof in self.args:
s = sem(x, axis, ddof)
self.assert_(np.issubdtype(s.dtype, np.floating))
self.assertIsInstance(s, (Series, np.ndarray))
def test_panel(self):
x = Panel(randn(4, 3, 2))
axes = self.axes + (2,)
args = itools.product(axes, self.ddof)
for axis, ddof in args:
s = sem(x, axis, ddof)
self.assertIsInstance(s, DataFrame)
def test_1d(self):
x = randn(2)
for axis, ddof in self.args:
if axis == 1:
self.assertRaises(IndexError, sem, x, axis, ddof)
else:
s = sem(x, axis, ddof)
dtype = s.dtype
self.assert_(np.issubdtype(dtype, np.floating))
def test_series(self):
x = Series(randn(13))
for axis, ddof in self.args:
if axis == 1:
self.assertRaises(IndexError, sem, x, axis, ddof)
else:
s = sem(x, axis, ddof)
try:
dtype = s.dtype
self.assert_(np.issubdtype(dtype, np.floating))
except AttributeError:
dtype = type(s)
self.assertIsInstance(s, (np.floating, float))
def test_2d(self):
x = randn(3, 2)
for axis, ddof in self.args:
s = sem(x, axis, ddof)
try:
dtype = s.dtype
except AttributeError:
dtype = type(s)
sshape = list(s.shape)
not_xshape = list(filter(lambda a: a != x.shape[axis], x.shape))
self.assertListEqual(not_xshape, sshape)
self.assert_(np.issubdtype(dtype, np.floating))
self.assertIsInstance(s, np.ndarray)
def test_3d(self):
x = randn(4, 3, 2)
axes = self.axes + (2,)
args = itools.product(axes, self.ddof)
for axis, ddof in args:
s = sem(x, axis, ddof)
try:
dtype = s.dtype
except AttributeError:
dtype = type(s)
self.assert_(np.issubdtype(dtype, np.floating))
sshape = list(s.shape)
not_xshape = list(filter(lambda a: a != x.shape[axis], x.shape))
self.assertListEqual(not_xshape, sshape)
self.assertIsInstance(s, np.ndarray)
def test_0d(self):
x = randn()
for axis, ddof in self.args:
s = sem(x, axis, ddof)
self.assertEqual(s, 0.0)
