-
Notifications
You must be signed in to change notification settings - Fork 63
/
Copy pathtest_table.py
33 lines (27 loc) · 927 Bytes
/
test_table.py
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
from __future__ import print_function
import os
import sys
def run_test():
print("Test mesh module")
Nvert = 6; Nelem = 8; Nbelem=2
mesh = mfem.Mesh(2, Nvert, Nelem, 2, 3)
tri_v = [[1., 0., 0.], [0., 1., 0.], [-1., 0., 0.],
[0., -1., 0.], [0., 0., 1.], [ 0., 0., -1.]]
tri_e = [[0, 1, 4], [1, 2, 4], [2, 3, 4], [3, 0, 4],
[1, 0, 5], [2, 1, 5], [3, 2, 5], [0, 3, 5]]
tri_l = [[1,4], [1,2]]
for j in range(Nvert):
mesh.AddVertex(tri_v[j])
for j in range(Nelem):
mesh.AddTriangle(tri_e[j], 1)
for j in range(Nbelem):
mesh.AddBdrSegment(tri_l[j], 1)
mesh.FinalizeTriMesh(1,1, True)
t = mesh.GetEdgeVertexTable()
t.Print()
if __name__=='__main__':
if len(sys.argv) > 1 and sys.argv[1] == '-p':
import mfem.par as mfem
else:
import mfem.ser as mfem
run_test()