1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
# vim:set ts=4 sw=4 sts=4 et:
import unittest
from igraph import *
class SpectralTests(unittest.TestCase):
def assertAlmostEqualMatrix(self, mat1, mat2, eps = 1e-7):
self.assertTrue(all(
abs(obs-exp) < eps
for obs, exp in zip(sum(mat1, []), sum(mat2, []))
))
def testLaplacian(self):
g=Graph.Full(3)
g.es["weight"] = [1, 2, 3]
self.assertTrue(g.laplacian() == [[ 2, -1, -1],\
[-1, 2, -1],\
[-1, -1, 2]])
self.assertAlmostEqualMatrix(g.laplacian(normalized=True),
[[1, -0.5, -0.5], [-0.5, 1, -0.5], [-0.5, -0.5, 1]])
mx0 = [[1., -1/(12**0.5), -2/(15**0.5)],
[-1/(12**0.5), 1., -3/(20**0.5)],
[-2/(15**0.5), -3/(20**0.5), 1.]]
self.assertAlmostEqualMatrix(g.laplacian("weight", True), mx0)
g=Graph.Tree(5, 2)
g.add_vertices(1)
self.assertTrue(g.laplacian() == [[ 2, -1, -1, 0, 0, 0],\
[-1, 3, 0, -1, -1, 0],\
[-1, 0, 1, 0, 0, 0],\
[ 0, -1, 0, 1, 0, 0],\
[ 0, -1, 0, 0, 1, 0],\
[ 0, 0, 0, 0, 0, 0]])
def suite():
spectral_suite = unittest.makeSuite(SpectralTests)
return unittest.TestSuite([spectral_suite])
def test():
runner = unittest.TextTestRunner()
runner.run(suite())
if __name__ == "__main__":
test()
|
