def test_1d_array(self):
x = randn(5)
axes = self.axes
arg_sets = itools.product(self.alphas, self.includes, axes)
for arg_set in arg_sets:
alpha, include, axis = arg_set
m = trimmean(x, alpha, include, axis)
self.assertIsInstance(m, float)
def test_0d_array(self):
x = np.asanyarray(randn())
axes = self.axes
arg_sets = itools.product(self.alphas, self.includes, axes)
for alpha, include, axis in arg_sets:
m = trimmean(x, alpha, include, axis)
self.assertIsInstance(m, float)
self.assertEqual(x, m)
def test_number(self):
x = float(randn())
axes = self.axes
arg_sets = itools.product(self.alphas, self.includes, axes)
for arg_set in arg_sets:
alpha, include, axis = arg_set
m = trimmean(x, alpha, include, axis)
self.assertIsInstance(m, float)
self.assertEqual(x, m)
def test_dataframe(self):
x = DataFrame(randn(3, 2))
axes = self.axes + (1,)
arg_sets = itools.product(self.alphas, self.includes, axes)
for arg_set in arg_sets:
alpha, include, axis = arg_set
m = trimmean(x, alpha, include, axis)
if axis is not None:
self.assertEqual(m.ndim, x.ndim - 1)
else:
self.assertIsInstance(m, float)
def test_3d_array(self):
x = randn(4, 3, 2)
axes = self.axes + (1, 2)
arg_sets = itools.product(self.alphas, self.includes, axes)
for arg_set in arg_sets:
alpha, include, axis = arg_set
if axis is not None:
self.assertRaises(Exception, trimmean, x, alpha, include, axis)
else:
m = trimmean(x, alpha, include, axis)
self.assertIsInstance(m, float)
