← Back to branch summary

~johan-hake/dolfin/general-rk-solver

« back to all changes in this revision

Viewing changes to test/unit/io/python/vtk.py

  • Committer: Johan Hake
  • Date: 2013-03-27 15:18:08 UTC
  • mfrom: (7352.1.227 working)
  • Revision ID: hake.dev@gmail.com-20130327151808-b73d4pueq1n432hg
Merge with trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
test/unit/io/python/vtk.py
25
25
 
26
26
# VTK file options
27
27
file_options = ["ascii", "base64", "compressed"]
 
28
mesh_functions = [CellFunction, FacetFunction, FaceFunction, EdgeFunction, VertexFunction]
 
29
mesh_function_types = ["size_t", "int", "double", "bool"]
 
30
 
 
31
class VTK_MeshFunction_Output(unittest.TestCase):
 
32
    """Test output of MeshFunctions to VTK files"""
 
33
    def test_save_1d_meshfunctions(self):
 
34
        #if MPI.num_processes() == 1:
 
35
        mesh = UnitIntervalMesh(32)
 
36
        for F in mesh_functions:
 
37
            if F in [FaceFunction, EdgeFunction]: continue
 
38
            for t in mesh_function_types:
 
39
                mf = F(t, mesh, 1)
 
40
                File("mf.pvd") << mf
 
41
                f = File("mf.pvd")
 
42
                f << (mf, 0.)
 
43
                f << (mf, 1.)
 
44
                for file_option in file_options:
 
45
                    File("mf.pvd", file_option) << mf
 
46
 
 
47
    def test_save_2d_meshfunctions(self):
 
48
        mesh = UnitSquareMesh(32, 32)
 
49
        for F in mesh_functions:
 
50
            for t in mesh_function_types:
 
51
                mf = F(t, mesh, 1)
 
52
                File("mf.pvd") << mf
 
53
                f = File("mf.pvd")
 
54
                f << (mf, 0.)
 
55
                f << (mf, 1.)
 
56
                for file_option in file_options:
 
57
                    File("mf.pvd", file_option) << mf
 
58
 
 
59
    def test_save_3d_meshfunctions(self):
 
60
        mesh = UnitCubeMesh(8, 8, 8)
 
61
        for F in mesh_functions:
 
62
            for t in mesh_function_types:
 
63
                mf = F(t, mesh, 1)
 
64
                File("mf.pvd") << mf
 
65
                f = File("mf.pvd")
 
66
                f << (mf, 0.)
 
67
                f << (mf, 1.)
 
68
                for file_option in file_options:
 
69
                    File("mf.pvd", file_option) << mf
28
70
 
29
71
class VTK_Mesh_Output(unittest.TestCase):
30
72
    """Test output of Meshes to VTK files"""
31
73
 
32
74
    def test_save_1d_mesh(self):
33
 
        if MPI.num_processes() == 1:
34
 
            mesh = UnitIntervalMesh(32)
35
 
            File("mesh.pvd") << mesh
36
 
            for file_option in file_options:
37
 
                File("mesh.pvd", file_option) << mesh
 
75
        mesh = UnitIntervalMesh(32)
 
76
        File("mesh.pvd") << mesh
 
77
        f = File("mesh.pvd")
 
78
        f << (mesh, 0.)
 
79
        f << (mesh, 1.)
 
80
        for file_option in file_options:
 
81
            File("mesh.pvd", file_option) << mesh
38
82
 
39
83
    def test_save_2d_mesh(self):
40
84
        mesh = UnitSquareMesh(32, 32)
41
85
        File("mesh.pvd") << mesh
 
86
        f = File("mesh.pvd")
 
87
        f << (mesh, 0.)
 
88
        f << (mesh, 1.)
42
89
        for file_option in file_options:
43
90
            File("mesh.pvd", file_option) << mesh
44
91
 
45
92
    def test_save_3d_mesh(self):
46
93
        mesh = UnitCubeMesh(8, 8, 8)
47
94
        File("mesh.pvd") << mesh
 
95
        f = File("mesh.pvd")
 
96
        f << (mesh, 0.)
 
97
        f << (mesh, 1.)
48
98
        for file_option in file_options:
49
99
            File("mesh.pvd", file_option) << mesh
50
100
 
58
108
            u = Function(FunctionSpace(mesh, "Lagrange", 2))
59
109
            u.vector()[:] = 1.0
60
110
            File("u.pvd") << u
 
111
            f = File("u.pvd")
 
112
            f << (u, 0.)
 
113
            f << (u, 1.)
61
114
            for file_option in file_options:
62
115
                File("u.pvd", file_option) << u
63
116
 
66
119
        u = Function(FunctionSpace(mesh, "Lagrange", 2))
67
120
        u.vector()[:] = 1.0
68
121
        File("u.pvd") << u
 
122
        f = File("u.pvd")
 
123
        f << (u, 0.)
 
124
        f << (u, 1.)
69
125
        for file_option in file_options:
70
126
            File("u.pvd", file_option) << u
71
127
 
74
130
        u = Function(FunctionSpace(mesh, "Lagrange", 2))
75
131
        u.vector()[:] = 1.0
76
132
        File("u.pvd") << u
 
133
        f = File("u.pvd")
 
134
        f << (u, 0.)
 
135
        f << (u, 1.)
77
136
        for file_option in file_options:
78
137
            File("u.pvd", file_option) << u
79
138
 
92
151
        u = Function(VectorFunctionSpace(mesh, "Lagrange", 2))
93
152
        u.vector()[:] = 1.0
94
153
        File("u.pvd") << u
 
154
        f = File("u.pvd")
 
155
        f << (u, 0.)
 
156
        f << (u, 1.)
95
157
        for file_option in file_options:
96
158
            File("u.pvd", file_option) << u
97
159
 
100
162
        u = Function(VectorFunctionSpace(mesh, "Lagrange", 2))
101
163
        u.vector()[:] = 1.0
102
164
        File("u.pvd") << u
 
165
        f = File("u.pvd")
 
166
        f << (u, 0.)
 
167
        f << (u, 1.)
103
168
        for file_option in file_options:
104
169
            File("u.pvd", file_option) << u
105
170
 
118
183
        u = Function(TensorFunctionSpace(mesh, "Lagrange", 2))
119
184
        u.vector()[:] = 1.0
120
185
        File("u.pvd") << u
 
186
        f = File("u.pvd")
 
187
        f << (u, 0.)
 
188
        f << (u, 1.)
121
189
        for file_option in file_options:
122
190
            File("u.pvd", file_option) << u
123
191
 
126
194
        u = Function(TensorFunctionSpace(mesh, "Lagrange", 2))
127
195
        u.vector()[:] = 1.0
128
196
        File("u.pvd") << u
 
197
        f = File("u.pvd")
 
198
        f << (u, 0.)
 
199
        f << (u, 1.)
129
200
        for file_option in file_options:
130
201
            File("u.pvd", file_option) << u
131
202