Skip to content

Commit 84eb186

Browse files
committed
Added test for parallel clustering
1 parent 00135a3 commit 84eb186

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

tests/test_unit_core.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,26 @@ def test_two_connected_clusters(self):
116116
ccs2 = mapper_connected_components(data, data, cover, clustering)
117117
self.assertEqual(len(data), len(ccs2))
118118

119+
def test_two_connected_clusters_parallel(self):
120+
data = [
121+
np.array([0.0, 1.0]), np.array([1.0, 0.0]),
122+
np.array([0.0, 0.0]), np.array([1.0, 1.0])]
123+
cover = BallCover(1.1, metric=dist)
124+
clustering = TrivialClustering()
125+
mp = MapperAlgorithm(
126+
cover=cover,
127+
clustering=clustering,
128+
n_jobs=4,
129+
)
130+
g = mp.fit_transform(data, data)
131+
self.assertEqual(2, len(g))
132+
for node in g.nodes():
133+
self.assertEqual(1, len(list(g.neighbors(node))))
134+
ccs = list(nx.connected_components(g))
135+
self.assertEqual(1, len(ccs))
136+
ccs2 = mapper_connected_components(data, data, cover, clustering)
137+
self.assertEqual(len(data), len(ccs2))
138+
119139
def test_connected_components(self):
120140
data = [0, 1, 2, 3]
121141

0 commit comments

Comments
 (0)