def test_labeled_blockmodel(): g = nx.DiGraph() g.add_edge(0,1) g.add_edge(0,2) g.add_edge(0,3) g.add_edge(0,4) p = {'B': [1,2,3,4], 'A': [0]} bg = utils.labeled_blockmodel(g,p) assert list(bg.edges(data=True))[0][2]['weight'] == 4.0, \ "Incorrect edge weight in labeled blockmodel" assert list(bg.edges()) == [('A','B')], \ "Incorrected edges in labeled blockmodel"
def test_labeled_blockmodel(): g = nx.DiGraph() g.add_edge(0, 1) g.add_edge(0, 2) g.add_edge(0, 3) g.add_edge(0, 4) p = {'B': [1, 2, 3, 4], 'A': [0]} bg = utils.labeled_blockmodel(g, p) assert bg.edges(data=True)[0][2]['weight'] == 4.0, \ "Incorrect edge weight in labeled blockmodel" assert bg.edges() == [('A','B')], \ "Incorrected edges in labeled blockmodel"
def test_labeled_blockmodel(self): g = nx.DiGraph() g.add_edge(0, 1) g.add_edge(0, 2) g.add_edge(0, 3) g.add_edge(0, 4) p = {"B": [1, 2, 3, 4], "A": [0]} bg = utils.labeled_blockmodel(g, p) self.assertTrue( list(bg.edges(data=True))[0][2]["weight"] == 4.0, msg="Incorrect edge weight in labeled blockmodel", ) self.assertTrue( list(bg.edges()) == [("A", "B")], msg="Incorrected edges in labeled blockmodel", )