def test_snowball_dirnet(self):
net = pynet.Net()
net[0][1] = 1
net[0][2] = 2
net[1][2] = 12
net[2][1] = 21
net[2][3] = 23
net[4][3] = 43
sball = transforms.snowball(net, 0, 1)
self.assertTrue(len(sball) == 3)
self.assertTrue(len([1 for i, j, w in sball.edges if w > 0]) == 2)
self.assertTrue(isinstance(sball, type(net)))
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
sball = transforms.snowball(net, 0, 1, True)
self.assertTrue(len(sball) == 3)
self.assertTrue(len([1 for i, j, w in sball.edges if w > 0]) == 4)
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
sball = transforms.snowball(net, 3, 2, False)
self.assertTrue(len(sball) == 5)
self.assertTrue(len([1 for i, j, w in sball.edges if w > 0]) == 5)
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
sball = transforms.snowball(net, 3, 2, True)
self.assertTrue(len(sball) == 5)
self.assertTrue(len([1 for i, j, w in sball.edges if w > 0]) == 6)
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
def test_snowball_dirnet(self): net = pynet.Net() net[0][1] = 1 net[0][2] = 2 net[1][2] = 12 net[2][1] = 21 net[2][3] = 23 net[4][3] = 43 sball = transforms.snowball(net, 0, 1) self.assertTrue(len(sball) == 3) self.assertTrue(len([1 for i,j,w in sball.edges if w > 0]) == 2) self.assertTrue(isinstance(sball, type(net))) for i,j,w in sball.edges: self.assertTrue(w == net[i][j]) sball = transforms.snowball(net, 0, 1, True) self.assertTrue(len(sball) == 3) self.assertTrue(len([1 for i,j,w in sball.edges if w > 0]) == 4) for i,j,w in sball.edges: self.assertTrue(w == net[i][j]) sball = transforms.snowball(net, 3, 2, False) self.assertTrue(len(sball) == 5) self.assertTrue(len([1 for i,j,w in sball.edges if w > 0]) == 5) for i,j,w in sball.edges: self.assertTrue(w == net[i][j]) sball = transforms.snowball(net, 3, 2, True) self.assertTrue(len(sball) == 5) self.assertTrue(len([1 for i,j,w in sball.edges if w > 0]) == 6) for i,j,w in sball.edges: self.assertTrue(w == net[i][j])
def test_snowball_symmnet(self):
net = pynet.SymmNet()
net[0][1] = 1
net[0][2] = 2
net[1][2] = 12
net[2][3] = 23
net[3][4] = 34
sball = transforms.snowball(net, 0, 1)
self.assertTrue(len(sball) == 3)
self.assertTrue(len(list(sball.edges)) == 2)
self.assertTrue(isinstance(sball, type(net)))
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
sball = transforms.snowball(net, 0, 1, True)
self.assertTrue(len(list(sball.edges)) == 3)
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
sball = transforms.snowball(net, 3, 2, False)
self.assertTrue(len(sball) == 5)
self.assertTrue(len(list(sball.edges)) == 4)
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
sball = transforms.snowball(net, 3, 2, True)
self.assertTrue(len(sball) == 5)
self.assertTrue(len(list(sball.edges)) == 5)
for i, j, w in sball.edges:
self.assertTrue(w == net[i][j])
def test_snowball_symmnet(self): net = pynet.SymmNet() net[0][1] = 1 net[0][2] = 2 net[1][2] = 12 net[2][3] = 23 net[3][4] = 34 sball = transforms.snowball(net, 0, 1) self.assertTrue(len(sball) == 3) self.assertTrue(len(list(sball.edges)) == 2) self.assertTrue(isinstance(sball, type(net))) for i,j,w in sball.edges: self.assertTrue(w == net[i][j]) sball = transforms.snowball(net, 0, 1, True) self.assertTrue(len(list(sball.edges)) == 3) for i,j,w in sball.edges: self.assertTrue(w == net[i][j]) sball = transforms.snowball(net, 3, 2, False) self.assertTrue(len(sball) == 5) self.assertTrue(len(list(sball.edges)) == 4) for i,j,w in sball.edges: self.assertTrue(w == net[i][j]) sball = transforms.snowball(net, 3, 2, True) self.assertTrue(len(sball) == 5) self.assertTrue(len(list(sball.edges)) == 5) for i,j,w in sball.edges: self.assertTrue(w == net[i][j])