def test_lemon(self):
g = get_string_graph()
seeds = ["$0$", "$2$", "$3$"]
com = algorithms.lemon(g, seeds, min_com_size=10, max_com_size=50)
self.assertEqual(type(com.communities), list)
if len(com.communities) > 0:
self.assertEqual(type(com.communities[0]), list)
self.assertEqual(type(com.communities[0][0]), str)
g = nx.karate_club_graph()
seeds = [0, 2, 3]
com = algorithms.lemon(g, seeds, min_com_size=10, max_com_size=50)
self.assertEqual(type(com.communities), list)
if len(com.communities) > 0:
self.assertEqual(type(com.communities[0]), list)
self.assertEqual(type(com.communities[0][0]), int)
def lemon(seeds, min_com_size, max_com_size) : return lambda G : algorithms.lemon(G, seeds=seeds, min_com_size=min_com_size, max_com_size=max_com_size) seeds = [1, 2, 3]