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])