def test_with_perfect_positive_correlation(self):
score = _calculate_pearson_correlation_score({
1: {'a': 2, 'b': 2},
2: {'a': 1, 'b': 1},
3: {'a': -1, 'b': -1},
})
self.assertEqual(1, score)
def test_with_good_correlation(self):
score = _calculate_pearson_correlation_score({
1: {'a': 2, 'b': 3},
2: {'a': 2, 'b': 2},
3: {'a': 0, 'b': 1},
})
self.assertTrue(score > 0.75)
def test_with_perfect_negative_correlation(self):
score = _calculate_pearson_correlation_score({
1: {'a': 2, 'b': -2},
2: {'a': 0, 'b': 0},
3: {'a': -2, 'b': 2},
})
self.assertEqual(-1, score)
def test_with_no_overlap(self):
score = _calculate_pearson_correlation_score({})
self.assertEqual(0, score)
