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
|
import unittest
from igraph import *
class IteratorTests(unittest.TestCase):
def testBFS(self):
g=Graph.Tree(10, 2)
vs=[v.index for v in g.bfsiter(0)]
self.assertEqual(vs, [0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
vs=[(v.index,dist,parent) for v,dist,parent in g.bfsiter(0, advanced=True)]
vs=[(v,d,p.index) for v,d,p in vs if p != None]
self.assertEqual(vs, [(1,1,0), (2,1,0), (3,2,1), (4,2,1), \
(5,2,2), (6,2,2), (7,3,3), (8,3,3), (9,3,4)])
def suite():
iterator_suite = unittest.makeSuite(IteratorTests)
return unittest.TestSuite([iterator_suite])
def test():
runner = unittest.TextTestRunner()
runner.run(suite())
if __name__ == "__main__":
test()
|
