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",
)