def test_em(self): g = get_string_graph() coms = algorithms.em(g, k=3) self.assertEqual(type(coms.communities), list) if len(coms.communities) > 0: self.assertEqual(type(coms.communities[0]), list) self.assertEqual(type(coms.communities[0][0]), str) g = nx.karate_club_graph() coms = algorithms.em(g, k=3) self.assertEqual(type(coms.communities), list) if len(coms.communities) > 0: self.assertEqual(type(coms.communities[0]), list) self.assertEqual(type(coms.communities[0][0]), int)
def em(n_coms) : return lambda G : algorithms.em(G, n_coms) algos['em'] = [em(len(c)) for c in lfr_comms]