Skip to content

Commit 67f63bc

Browse files
committed
Added a unit-test for UCLUS linkage.
1 parent 53ab0b2 commit 67f63bc

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

test.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,27 @@ def testUCLUS(self):
137137
result = sorted([sorted(_) for _ in cl.getlevel(40)])
138138
self.assertEqual(result, expected)
139139

140+
def testAverageLinkage(self):
141+
cl = HierarchicalClustering(self.__data,
142+
lambda x, y: abs(x - y),
143+
linkage='average')
144+
# TODO: The current test-data does not really trigger a difference
145+
# between UCLUS and "average" linkage.
146+
expected = [
147+
[24],
148+
[84],
149+
[124, 131, 134],
150+
[336, 365, 365, 391, 398],
151+
[518, 542, 564],
152+
[594],
153+
[676],
154+
[791],
155+
[835],
156+
[940, 956, 971],
157+
]
158+
result = sorted([sorted(_) for _ in cl.getlevel(40)])
159+
self.assertEqual(result, expected)
160+
140161
def testUnmodifiedData(self):
141162
cl = HierarchicalClustering(self.__data, lambda x, y: abs(x - y))
142163
new_data = []

0 commit comments

Comments
 (0)