(self)
| 11 | class TestGraphWriterReader(Testing.vtkTest): |
| 12 | |
| 13 | def test(self): |
| 14 | |
| 15 | mdgTree = dm.vtkMutableDirectedGraph() |
| 16 | mug = dm.vtkMutableUndirectedGraph() |
| 17 | |
| 18 | for i in range(10): |
| 19 | mdgTree.AddVertex() |
| 20 | mug.AddVertex() |
| 21 | |
| 22 | mdgTree.AddEdge(0, 1) |
| 23 | mdgTree.AddEdge(0, 2) |
| 24 | mdgTree.AddEdge(0, 3) |
| 25 | mdgTree.AddEdge(1, 4) |
| 26 | mdgTree.AddEdge(1, 5) |
| 27 | mdgTree.AddEdge(2, 6) |
| 28 | mdgTree.AddEdge(2, 7) |
| 29 | mdgTree.AddEdge(3, 8) |
| 30 | mdgTree.AddEdge(3, 9) |
| 31 | |
| 32 | # Undirected graph with parallel edges |
| 33 | # and self-loops. |
| 34 | mug.AddEdge(0, 0) |
| 35 | mug.AddEdge(0, 1) |
| 36 | mug.AddEdge(1, 0) |
| 37 | mug.AddEdge(1, 2) |
| 38 | mug.AddEdge(1, 3) |
| 39 | mug.AddEdge(4, 5) |
| 40 | mug.AddEdge(4, 5) |
| 41 | mug.AddEdge(6, 7) |
| 42 | mug.AddEdge(7, 7) |
| 43 | |
| 44 | w = il.vtkGraphWriter() |
| 45 | |
| 46 | tmpdir = vtkGetTempDir() |
| 47 | fname = tmpdir+"/testgrapwriread.vtk" |
| 48 | |
| 49 | w.SetFileName(fname) |
| 50 | w.SetInputData(mdgTree) |
| 51 | w.Write() |
| 52 | |
| 53 | r = il.vtkGraphReader() |
| 54 | r.SetFileName(fname) |
| 55 | r.Update() |
| 56 | |
| 57 | g = r.GetOutput() |
| 58 | |
| 59 | self.assertTrue(g.IsA("vtkDirectedGraph")) |
| 60 | self.assertTrue(g.GetNumberOfVertices() == 10) |
| 61 | self.assertTrue(g.GetNumberOfEdges() == 9) |
| 62 | self.assertTrue(g.GetTargetVertex(0) == 1) |
| 63 | |
| 64 | del(r) |
| 65 | import gc |
| 66 | gc.collect() |
| 67 | os.remove(fname) |
| 68 | |
| 69 | w.SetFileName(fname) |
| 70 | w.SetInputData(mug) |
nothing calls this directly
no test coverage detected