1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import unittest
from cluster.linkage import single, complete, uclus, average
class LinkageMethods(unittest.TestCase):
def setUp(self):
self.set_a = [1, 2, 3, 4]
self.set_b = [10, 11, 12, 13, 14, 15, 100]
self.dist = lambda x, y: abs(x-y) # NOQA
def test_single_distance(self):
result = single(self.set_a, self.set_b, self.dist)
expected = 6
self.assertEqual(result, expected)
def test_complete_distance(self):
result = complete(self.set_a, self.set_b, self.dist)
expected = 99
self.assertEqual(result, expected)
def test_uclus_distance(self):
result = uclus(self.set_a, self.set_b, self.dist)
expected = 10.5
self.assertEqual(result, expected)
def test_average_distance(self):
result = average(self.set_a, self.set_b, self.dist)
expected = 22.5
self.assertEqual(result, expected)
if __name__ == '__main__':
import logging
suite = unittest.TestSuite((
unittest.makeSuite(LinkageMethods),
))
logging.basicConfig(level=logging.DEBUG)
unittest.TextTestRunner(verbosity=2).run(suite)
|
