def test_get_vertex_coordinates_with_geo_ref(self):
x0 = 314036.58727982
y0 = 6224951.2960092
geo = Geo_reference(56, x0, y0)
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
nodes = num.array([a, b, c, d, e, f])
nodes_absolute = geo.get_absolute(nodes)
# bac, bce, ecf, dbe
triangles = num.array([[1, 0, 2], [1, 2, 4], [4, 2, 5], [3, 1, 4]],
num.int)
domain = General_mesh(nodes, triangles, geo_reference=geo)
verts = domain.get_vertex_coordinates(triangle_id=0) # bac
msg = ("num.array([b,a,c])=\n%s\nshould be close to 'verts'=\n%s" %
(str(num.array([b, a, c])), str(verts)))
self.assertTrue(num.allclose(num.array([b, a, c]), verts), msg)
verts = domain.get_vertex_coordinates(triangle_id=0)
msg = ("num.array([b,a,c])=\n%s\nshould be close to 'verts'=\n%s" %
(str(num.array([b, a, c])), str(verts)))
self.assertTrue(num.allclose(num.array([b, a, c]), verts), msg)
verts = domain.get_vertex_coordinates(triangle_id=0, absolute=True)
msg = ("num.array([...])=\n%s\nshould be close to 'verts'=\n%s" % (str(
num.array([
nodes_absolute[1], nodes_absolute[0], nodes_absolute[2]
])), str(verts)))
self.assertTrue(
num.allclose(
num.array(
[nodes_absolute[1], nodes_absolute[0], nodes_absolute[2]]),
verts), msg)
verts = domain.get_vertex_coordinates(triangle_id=0, absolute=True)
msg = ("num.array([...])=\n%s\nshould be close to 'verts'=\n%s" % (str(
num.array([
nodes_absolute[1], nodes_absolute[0], nodes_absolute[2]
])), str(verts)))
self.assertTrue(
num.allclose(
num.array(
[nodes_absolute[1], nodes_absolute[0], nodes_absolute[2]]),
verts), msg)
def test_get_vertex_coordinates_with_geo_ref(self):
x0 = 314036.58727982
y0 = 6224951.2960092
geo = Geo_reference(56, x0, y0)
a = [0.0, 0.0]
b = [0.0, 2.0]
c = [2.0, 0.0]
d = [0.0, 4.0]
e = [2.0, 2.0]
f = [4.0, 0.0]
nodes = num.array([a, b, c, d, e, f])
nodes_absolute = geo.get_absolute(nodes)
# bac, bce, ecf, dbe
triangles = num.array([[1,0,2], [1,2,4], [4,2,5], [3,1,4]], num.int)
domain = General_mesh(nodes, triangles, geo_reference=geo)
verts = domain.get_vertex_coordinates(triangle_id=0) # bac
msg = ("num.array([b,a,c])=\n%s\nshould be close to 'verts'=\n%s"
% (str(num.array([b,a,c])), str(verts)))
self.assertTrue(num.allclose(num.array([b,a,c]), verts), msg)
verts = domain.get_vertex_coordinates(triangle_id=0)
msg = ("num.array([b,a,c])=\n%s\nshould be close to 'verts'=\n%s"
% (str(num.array([b,a,c])), str(verts)))
self.assert_(num.allclose(num.array([b,a,c]), verts), msg)
verts = domain.get_vertex_coordinates(triangle_id=0, absolute=True)
msg = ("num.array([...])=\n%s\nshould be close to 'verts'=\n%s"
% (str(num.array([nodes_absolute[1],
nodes_absolute[0],
nodes_absolute[2]])),
str(verts)))
self.assert_(num.allclose(num.array([nodes_absolute[1],
nodes_absolute[0],
nodes_absolute[2]]),
verts), msg)
verts = domain.get_vertex_coordinates(triangle_id=0,
absolute=True)
msg = ("num.array([...])=\n%s\nshould be close to 'verts'=\n%s"
% (str(num.array([nodes_absolute[1],
nodes_absolute[0],
nodes_absolute[2]])),
str(verts)))
self.assert_(num.allclose(num.array([nodes_absolute[1],
nodes_absolute[0],
nodes_absolute[2]]),
verts), msg)
def test_get_vertex_coordinates(self):
from anuga.abstract_2d_finite_volumes.mesh_factory import rectangular
#Create basic mesh
nodes, triangles, _ = rectangular(1, 3)
domain = General_mesh(nodes, triangles)
assert num.allclose(domain.get_nodes(), nodes)
M = domain.number_of_triangles
V = domain.get_vertex_coordinates()
assert V.shape[0] == 3 * M
for i in range(M):
for j in range(3):
k = triangles[i, j] #Index of vertex j in triangle i
assert num.allclose(V[3 * i + j, :], nodes[k])
def test_get_vertex_coordinates(self):
from anuga.abstract_2d_finite_volumes.mesh_factory import rectangular
#Create basic mesh
nodes, triangles, _ = rectangular(1, 3)
domain = General_mesh(nodes, triangles)
assert num.allclose(domain.get_nodes(), nodes)
M = domain.number_of_triangles
V = domain.get_vertex_coordinates()
assert V.shape[0] == 3*M
for i in range(M):
for j in range(3):
k = triangles[i,j] #Index of vertex j in triangle i
assert num.allclose(V[3*i+j,:], nodes[k])
def test_get_vertex_coordinates_triangle_id(self):
"""test_get_vertex_coordinates_triangle_id
Test that vertices for one triangle can be returned.
"""
from anuga.abstract_2d_finite_volumes.mesh_factory import rectangular
#Create basic mesh
nodes, triangles, _ = rectangular(1, 3)
domain = General_mesh(nodes, triangles)
assert num.allclose(domain.get_nodes(), nodes)
M = domain.number_of_triangles
for i in range(M):
V = domain.get_vertex_coordinates(triangle_id=i)
assert V.shape[0] == 3
for j in range(3):
k = triangles[i, j] #Index of vertex j in triangle i
assert num.allclose(V[j, :], nodes[k])
def test_get_vertex_coordinates_triangle_id(self):
"""test_get_vertex_coordinates_triangle_id
Test that vertices for one triangle can be returned.
"""
from anuga.abstract_2d_finite_volumes.mesh_factory import rectangular
#Create basic mesh
nodes, triangles, _ = rectangular(1, 3)
domain = General_mesh(nodes, triangles)
assert num.allclose(domain.get_nodes(), nodes)
M = domain.number_of_triangles
for i in range(M):
V = domain.get_vertex_coordinates(triangle_id=i)
assert V.shape[0] == 3
for j in range(3):
k = triangles[i,j] #Index of vertex j in triangle i
assert num.allclose(V[j,:], nodes[k])
