Skip to content

Commit 53ab0b2

Browse files
committed
Added unit-test for uclus linkage.
1 parent f800986 commit 53ab0b2

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

test.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,25 @@ def testCompleteLinkage(self):
117117
]
118118
self.assertEqual(result, expected)
119119

120+
def testUCLUS(self):
121+
"Basic Hierarchical Clustering test with integers"
122+
cl = HierarchicalClustering(self.__data,
123+
lambda x, y: abs(x - y),
124+
linkage='uclus')
125+
expected = [
126+
[24],
127+
[84],
128+
[124, 131, 134],
129+
[336, 365, 365, 391, 398],
130+
[518, 542, 564],
131+
[594],
132+
[676],
133+
[791],
134+
[835],
135+
[940, 956, 971],
136+
]
137+
result = sorted([sorted(_) for _ in cl.getlevel(40)])
138+
self.assertEqual(result, expected)
120139

121140
def testUnmodifiedData(self):
122141
cl = HierarchicalClustering(self.__data, lambda x, y: abs(x - y))

0 commit comments

Comments
 (0)