def test_wCommunities(self):
g = get_string_graph()
nx.set_edge_attributes(g, values=1, name='weight')
communities = algorithms.wCommunity(g,
min_bel_degree=0.6,
threshold_bel_degree=0.6)
self.assertEqual(type(communities.communities), list)
if len(communities.communities) > 0:
self.assertEqual(type(communities.communities[0]), list)
if len(communities.communities[0]) > 0:
self.assertEqual(type(communities.communities[0][0]), int)
g = nx.karate_club_graph()
nx.set_edge_attributes(g, values=1, name='weight')
communities = algorithms.wCommunity(g,
min_bel_degree=0.6,
threshold_bel_degree=0.6)
self.assertEqual(type(communities.communities), list)
if len(communities.communities) > 0:
self.assertEqual(type(communities.communities[0]), list)
if len(communities.communities[0]) > 0:
self.assertEqual(type(communities.communities[0][0]), int)
elif (options.method == 'danmf'):
communities = algorithms.danmf(g, **clust_kwargs)
elif (options.method == 'ego_networks'):
communities = algorithms.ego_networks(g, **clust_kwargs)
elif (options.method == 'egonet_splitter'):
communities = algorithms.egonet_splitter(g, **clust_kwargs)
elif (options.method == 'nmnf'):
communities = algorithms.nmnf(g, **clust_kwargs)
elif (options.method == 'nnsed'):
communities = algorithms.nnsed(g, **clust_kwargs)
elif (options.method == 'slpa'):
communities = algorithms.slpa(g, **clust_kwargs)
elif (options.method == 'bimlpa'):
communities = actlgorithms.bimlpa(g, **clust_kwargs)
elif (options.method == 'wcommunity'):
communities = algorithms.wCommunity(g, **clust_kwargs)
elif (options.method == 'aslpaw'):
import warnings
with warnings.catch_warnings():
warnings.filterwarnings("ignore")
communities = algorithms.aslpaw(g)
elif (options.method == 'external'):
from collections import defaultdict
from cdlib import NodeClustering
coms_to_node = defaultdict(list)
f = open(options.external, 'r')
for line in f:
fields = line.strip("\n").split("\t")
coms_to_node[fields[0]].append(fields[1])
coms = [list(c) for c in coms_to_node.values()]
communities = NodeClustering(coms,