def test_convert_to_block_df(self): a = np.array([[0, 0, 4], [1, 0, 1], [2, 0, 1], [0, 1, 3], [1, 1, 2], [2, 1, 2], [0, 2, 4], [1, 2, 3], [2, 2, 6], [0, 3, 4], [1, 3, 5], [2, 3, 4], [0, 4, 5], [1, 4, 6], [2, 4, 1], [0, 5, 6], [1, 5, 7], [2, 5, 5], [0, 6, 3], [1, 6, 7], [2, 6, 1]], dtype=np.float) result = sp.posthoc_nemenyi_friedman(a, y_col=2, group_col=1, block_col=0, melted=True)[0].values result2 = sp.posthoc_nemenyi_friedman(self.df_bn)[0].values self.assertTrue(np.allclose(result, result2))
def test_posthoc_nemenyi_friedman(self): p_results = np.array([[-1., 0.9, 0.82163255, 0.9, 0.9, 0.21477876, 0.9], [0.9, -1., 0.87719193, 0.9, 0.9, 0.25967965, 0.9], [0.82163255, 0.87719193, -1., 0.9, 0.9, 0.9, 0.9], [0.9, 0.9, 0.9, -1., 0.9, 0.87719193, 0.9], [0.9, 0.9, 0.9, 0.9, -1., 0.87719193, 0.9], [0.21477876, 0.25967965, 0.9, 0.87719193, 0.87719193, -1., 0.54381888], [0.9, 0.9, 0.9, 0.9, 0.9, 0.54381888, -1.]]) results = sp.posthoc_nemenyi_friedman(self.df_bn) self.assertTrue(np.allclose(results, p_results))
def test_convert_to_block_df(self): a = np.array( [[0, 0, 4], [1, 0, 1], [2, 0, 1], [0, 1, 3], [1, 1, 2], [2, 1, 2], [0, 2, 4], [1, 2, 3], [2, 2, 6], [0, 3, 4], [1, 3, 5], [2, 3, 4], [0, 4, 5], [1, 4, 6], [2, 4, 1], [0, 5, 6], [1, 5, 7], [2, 5, 5], [0, 6, 3], [1, 6, 7], [2, 6, 1]], dtype=float) df_a = DataFrame(a, columns=['blk_col', 'grp_col', 'y_col']) result = sp.posthoc_nemenyi_friedman(a, y_col=2, group_col=1, block_col=0, melted=True)[0].values result2 = sp.posthoc_nemenyi_friedman(self.df_bn)[0].values result3 = sp.posthoc_nemenyi_friedman(df_a, y_col='y_col', group_col='grp_col', block_col='blk_col', melted=True)[0].values self.assertTrue(np.allclose(result, result2)) self.assertTrue(np.allclose(result, result3)) self.assertTrue(np.allclose(result2, result3))