def test_posthoc_mannwhitney_ndarray(self):
_x = [[1,2,3,5,1], [12,31,54,50,40], [10,12,6,74,11]]
x = np.array(_x)
g = np.repeat([0,1,2], 5)
nd = np.column_stack((x.ravel(), g))
xdf = DataFrame(dict(zip(list('abc'), _x))).melt(var_name='groups', value_name='vals')
results = sp.posthoc_mannwhitney(xdf, val_col = 'vals', group_col = 'groups').values
nd_results = sp.posthoc_mannwhitney(nd, val_col=0, group_col=1).values
self.assertTrue(np.allclose(nd_results, results))
def test_posthoc_mannwhitney(self):
r_results = np.array([[-1, 3.420508e-08, 1.714393e-02],
[3.420508e-08, -1, 1.968352e-05],
[1.714393e-02, 1.968352e-05, -1]])
results = sp.posthoc_mannwhitney(self.df, val_col = 'pulse', group_col = 'kind').values
self.assertTrue(np.allclose(results, r_results))