def test_coeff_with_same_and_floats(self): row1 = [5.0, 4.0, 3.0, 2, 1] ranks1 = np.array([5, 4, 3, 2, 1]) ranksMean1 = ranks1 - 3 row2 = [1.0, 2.0, 1.0, 4.0, 5.0] ranks2 = np.array([1.5, 3, 1.5, 4, 5]) ranksMean2 = ranks2 - np.mean(ranks2) spear1_2 = spearman.findSpearmanRank(row1, row2) spear2_1 = spearman.findSpearmanRank(row2, row1) spearExpect = findCorrelationCoeff(ranksMean1, ranksMean2, True) self.nearly_same(spear1_2, spearExpect, absTol=1e-18, relTol=1e-12) self.nearly_same(spear2_1, spearExpect, absTol=1e-18, relTol=1e-12)
def test_coeff_basic(self): row1 = [1, 2, 3, 4, 5] ranks1 = np.array([1, 2, 3, 4, 5]) ranksMean1 = ranks1 - 3 row2 = [5.0, 4.0, 3.0, 2, 1] ranks2 = np.array([5, 4, 3, 2, 1]) ranksMean2 = ranks2 - 3 spear1_2 = spearman.findSpearmanRank(row1, row2) spear2_1 = spearman.findSpearmanRank(row2, row1) spearExpect = findCorrelationCoeff(ranksMean1, ranksMean2, True) self.assertTrue(spearExpect == -1) self.nearly_same(spear1_2, spearExpect, absTol=1e-18, relTol=1e-12) self.nearly_same(spear2_1, spearExpect, absTol=1e-18, relTol=1e-12)