Skip to content

Commit d47c5ab

Browse files
committed
Added unit test for multiprocessing.
1 parent 183fd68 commit d47c5ab

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,13 @@ def testUnmodifiedData(self):
164164
[new_data.extend(_) for _ in cl.getlevel(40)]
165165
self.assertEqual(sorted(new_data), sorted(self.__data))
166166

167+
def testMultiprocessing(self):
168+
cl = HierarchicalClustering(self.__data, lambda x, y: abs(x - y),
169+
num_processes=4)
170+
new_data = []
171+
[new_data.extend(_) for _ in cl.getlevel(40)]
172+
self.assertEqual(sorted(new_data), sorted(self.__data))
173+
167174

168175
class HClusterStringTestCase(unittest.TestCase):
169176

0 commit comments

Comments
 (0)