class TestMeshReadCoordinates(unittest.TestCase):
def _writeUnstructMesh(self,name,ndim,elem):
import numpy
import random
nnodes=random.randint(10,512)
mesh = self.fh.add_unstruct_mesh(name,elem)
x=numpy.random.random_sample((nnodes))
if ndim == 1:
mesh.write_coordinates(x)
return x
if ndim == 2:
y=numpy.random.random_sample((nnodes))
mesh.write_coordinates(x,y)
return (x,y)
else:
y=numpy.random.random_sample((nnodes))
z=numpy.random.random_sample((nnodes))
mesh.write_coordinates(x,y,z)
return (x,y,z)
def setUp(self):
import os
from Danu import Output
from Danu import UNSTRUCTURED_MESH, STRUCTURED_MESH
from Danu import LINE_ELEM, TRI_ELEM, QUAD_ELEM, TET_ELEM, HEX_ELEM
self.filename = 'test-Mesh.h5'
if os.path.exists(self.filename):
os.remove(self.filename)
self.fh = Output(self.filename,'w')
self.valid_mesh_types = [UNSTRUCTURED_MESH, STRUCTURED_MESH]
self.valid_mesh_elems1 = [LINE_ELEM]
self.valid_mesh_elems2 = [TRI_ELEM, QUAD_ELEM]
self.valid_mesh_elems3 = [TET_ELEM, HEX_ELEM]
self.valid_mesh_1D = []
self.valid_mesh_2D = []
self.valid_mesh_3D = []
for elem in self.valid_mesh_elems1:
mesh_name='Mesh 1D' + str(elem)
self._writeUnstructMesh(mesh_name,1,elem)
self.valid_mesh_1D.insert(0,mesh_name)
for elem in self.valid_mesh_elems2:
mesh_name='Mesh 2D' + str(elem)
self._writeUnstructMesh(mesh_name,2,elem)
self.valid_mesh_2D.insert(0,mesh_name)
for elem in self.valid_mesh_elems3:
mesh_name='Mesh 3D' + str(elem)
self._writeUnstructMesh(mesh_name,3,elem)
self.valid_mesh_3D.insert(0,mesh_name)
def tearDown(self):
if os.path.exists(self.filename):
self.fh.close()
os.remove(self.filename)
def test_read_coord_1D(self):
import numpy
for mesh_name in self.valid_mesh_1D:
mesh=self.fh.get_mesh(mesh_name)
x=mesh.read_coordinates(0,1024)
mesh=self.fh.get_mesh(self.valid_mesh_1D[0])
try:
mesh.read_coordiantes(1,1024)
except:
print 'Caught invalid dimension request'
def test_read_coord_2D(self):
import numpy
for mesh_name in self.valid_mesh_2D:
mesh=self.fh.get_mesh(mesh_name)
x=mesh.read_coordinates(0,1024)
y=mesh.read_coordinates(1,1024)
def test_read_coord_3D(self):
import numpy
for mesh_name in self.valid_mesh_3D:
mesh=self.fh.get_mesh(mesh_name)
x=mesh.read_coordinates(0,1024)
y=mesh.read_coordinates(1,1024)
z=mesh.read_coordinates(2,1024)
def test_read_fails(self):
import numpy
mesh=self.fh.get_mesh(self.valid_mesh_2D[0])
try:
mesh.read_coordinates(3,1024)
except:
print 'Caught invalid dimension request'
try:
mesh.read_coordinates(0,5)
except:
print 'Caught invalid nnodes size'
def test_coordinates_1D(self):
import numpy
mesh=self.fh.get_mesh(self.valid_mesh_1D[0])
coordinates=mesh.coordinates()
self.assertEqual(len(coordinates),1)
nnodes=coordinates[0].size
idx=0
for array in coordinates:
read_array=mesh.read_coordinates(idx,nnodes)
self.assertEqual(read_array.all(),array.all())
def test_coordinates_2D(self):
import numpy
mesh=self.fh.get_mesh(self.valid_mesh_2D[1])
coordinates=mesh.coordinates()
self.assertEqual(len(coordinates),2)
nnodes=coordinates[0].size
idx=0
for array in coordinates:
read_array=mesh.read_coordinates(idx,nnodes)
self.assertEqual(read_array.all(),array.all())
def test_coordinates_3D(self):
import numpy
mesh=self.fh.get_mesh(self.valid_mesh_3D[1])
coordinates=mesh.coordinates()
self.assertEqual(len(coordinates),3)
nnodes=coordinates[0].size
idx=0
for array in coordinates:
read_array=mesh.read_coordinates(idx,nnodes)
self.assertEqual(read_array.all(),array.all())
class TestMeshReadCoordinates(unittest.TestCase):
def _writeUnstructMesh(self, name, ndim, elem):
import numpy
import random
nnodes = random.randint(10, 512)
mesh = self.fh.add_unstruct_mesh(name, elem)
x = numpy.random.random_sample((nnodes))
if ndim == 1:
mesh.write_coordinates(x)
return x
if ndim == 2:
y = numpy.random.random_sample((nnodes))
mesh.write_coordinates(x, y)
return (x, y)
else:
y = numpy.random.random_sample((nnodes))
z = numpy.random.random_sample((nnodes))
mesh.write_coordinates(x, y, z)
return (x, y, z)
def setUp(self):
import os
from Danu import Output
from Danu import UNSTRUCTURED_MESH, STRUCTURED_MESH
from Danu import LINE_ELEM, TRI_ELEM, QUAD_ELEM, TET_ELEM, HEX_ELEM
self.filename = 'test-Mesh.h5'
if os.path.exists(self.filename):
os.remove(self.filename)
self.fh = Output(self.filename, 'w')
self.valid_mesh_types = [UNSTRUCTURED_MESH, STRUCTURED_MESH]
self.valid_mesh_elems1 = [LINE_ELEM]
self.valid_mesh_elems2 = [TRI_ELEM, QUAD_ELEM]
self.valid_mesh_elems3 = [TET_ELEM, HEX_ELEM]
self.valid_mesh_1D = []
self.valid_mesh_2D = []
self.valid_mesh_3D = []
for elem in self.valid_mesh_elems1:
mesh_name = 'Mesh 1D' + str(elem)
self._writeUnstructMesh(mesh_name, 1, elem)
self.valid_mesh_1D.insert(0, mesh_name)
for elem in self.valid_mesh_elems2:
mesh_name = 'Mesh 2D' + str(elem)
self._writeUnstructMesh(mesh_name, 2, elem)
self.valid_mesh_2D.insert(0, mesh_name)
for elem in self.valid_mesh_elems3:
mesh_name = 'Mesh 3D' + str(elem)
self._writeUnstructMesh(mesh_name, 3, elem)
self.valid_mesh_3D.insert(0, mesh_name)
def tearDown(self):
if os.path.exists(self.filename):
self.fh.close()
os.remove(self.filename)
def test_read_coord_1D(self):
import numpy
for mesh_name in self.valid_mesh_1D:
mesh = self.fh.get_mesh(mesh_name)
x = mesh.read_coordinates(0, 1024)
mesh = self.fh.get_mesh(self.valid_mesh_1D[0])
try:
mesh.read_coordiantes(1, 1024)
except:
print 'Caught invalid dimension request'
def test_read_coord_2D(self):
import numpy
for mesh_name in self.valid_mesh_2D:
mesh = self.fh.get_mesh(mesh_name)
x = mesh.read_coordinates(0, 1024)
y = mesh.read_coordinates(1, 1024)
def test_read_coord_3D(self):
import numpy
for mesh_name in self.valid_mesh_3D:
mesh = self.fh.get_mesh(mesh_name)
x = mesh.read_coordinates(0, 1024)
y = mesh.read_coordinates(1, 1024)
z = mesh.read_coordinates(2, 1024)
def test_read_fails(self):
import numpy
mesh = self.fh.get_mesh(self.valid_mesh_2D[0])
try:
mesh.read_coordinates(3, 1024)
except:
print 'Caught invalid dimension request'
try:
mesh.read_coordinates(0, 5)
except:
print 'Caught invalid nnodes size'
def test_coordinates_1D(self):
import numpy
mesh = self.fh.get_mesh(self.valid_mesh_1D[0])
coordinates = mesh.coordinates()
self.assertEqual(len(coordinates), 1)
nnodes = coordinates[0].size
idx = 0
for array in coordinates:
read_array = mesh.read_coordinates(idx, nnodes)
self.assertEqual(read_array.all(), array.all())
def test_coordinates_2D(self):
import numpy
mesh = self.fh.get_mesh(self.valid_mesh_2D[1])
coordinates = mesh.coordinates()
self.assertEqual(len(coordinates), 2)
nnodes = coordinates[0].size
idx = 0
for array in coordinates:
read_array = mesh.read_coordinates(idx, nnodes)
self.assertEqual(read_array.all(), array.all())
def test_coordinates_3D(self):
import numpy
mesh = self.fh.get_mesh(self.valid_mesh_3D[1])
coordinates = mesh.coordinates()
self.assertEqual(len(coordinates), 3)
nnodes = coordinates[0].size
idx = 0
for array in coordinates:
read_array = mesh.read_coordinates(idx, nnodes)
self.assertEqual(read_array.all(), array.all())
class TestMeshUtils(unittest.TestCase):
def setUp(self):
from Danu import Output
from Danu import UNSTRUCTURED_MESH, STRUCTURED_MESH
from Danu import LINE_ELEM, TRI_ELEM, QUAD_ELEM, TET_ELEM, HEX_ELEM
self.filename = 'test-Mesh.h5'
if os.path.exists(self.filename):
os.remove(self.filename)
self.fh = Output(self.filename,'w')
self.mesh_count = 0
self.mesh_names = []
self.valid_mesh_types = [UNSTRUCTURED_MESH, STRUCTURED_MESH]
self.valid_mesh_elems1 = [LINE_ELEM]
self.valid_mesh_elems2 = [TRI_ELEM, QUAD_ELEM]
self.valid_mesh_elems3 = [TET_ELEM, HEX_ELEM]
def tearDown(self):
if os.path.exists(self.filename):
self.fh.close()
os.remove(self.filename)
def test_mesh_open(self):
import Danu
mesh_name='Test Mesh DNE'
self.failUnlessRaises(RuntimeError,self.fh.get_mesh,mesh_name)
Mesh(self.fh,mesh_name,Danu.UNSTRUCTURED_MESH,Danu.LINE_ELEM)
mesh=self.fh.get_mesh(mesh_name)
def test_mesh_destory(self):
import Danu
mesh_name='Test Mesh'
mesh=Mesh(self.fh,mesh_name,Danu.UNSTRUCTURED_MESH,Danu.LINE_ELEM)
del mesh
def test_mesh_create_1D(self):
import random
import string
for mesh_type in self.valid_mesh_types:
for elem in self.valid_mesh_elems1:
mesh_name = ''.join(random.choice(string.letters) for i in xrange(16))
mesh = Mesh(self.fh,mesh_name,mesh_type,elem)
self.assertEqual(mesh.dim,1, 'Incorrect dimension')
self.assertEqual(mesh.mesh_type,mesh_type,'Incorrect mesh type')
self.assertEqual(mesh.elem_type,elem,'Incorrect mesh element type')
def test_mesh_create_2D(self):
import random
import string
for mesh_type in self.valid_mesh_types:
for elem in self.valid_mesh_elems2:
mesh_name = ''.join(random.choice(string.letters) for i in xrange(16))
mesh = Mesh(self.fh,mesh_name,mesh_type,elem)
self.assertEqual(mesh.dim,2, 'Incorrect dimension')
self.assertEqual(mesh.mesh_type,mesh_type,'Incorrect mesh type')
self.assertEqual(mesh.elem_type,elem,'Incorrect mesh element type')
def test_mesh_create_3D(self):
import random
import string
for mesh_type in self.valid_mesh_types:
for elem in self.valid_mesh_elems3:
mesh_name = ''.join(random.choice(string.letters) for i in xrange(16))
mesh = Mesh(self.fh,mesh_name,mesh_type,elem)
self.assertEqual(mesh.dim,3, 'Incorrect dimension')
self.assertEqual(mesh.mesh_type,mesh_type,'Incorrect mesh type')
self.assertEqual(mesh.elem_type,elem,'Incorrect mesh element type')
class TestMeshUtils(unittest.TestCase):
def setUp(self):
from Danu import Output
from Danu import UNSTRUCTURED_MESH, STRUCTURED_MESH
from Danu import LINE_ELEM, TRI_ELEM, QUAD_ELEM, TET_ELEM, HEX_ELEM
self.filename = 'test-Mesh.h5'
if os.path.exists(self.filename):
os.remove(self.filename)
self.fh = Output(self.filename, 'w')
self.mesh_count = 0
self.mesh_names = []
self.valid_mesh_types = [UNSTRUCTURED_MESH, STRUCTURED_MESH]
self.valid_mesh_elems1 = [LINE_ELEM]
self.valid_mesh_elems2 = [TRI_ELEM, QUAD_ELEM]
self.valid_mesh_elems3 = [TET_ELEM, HEX_ELEM]
def tearDown(self):
if os.path.exists(self.filename):
self.fh.close()
os.remove(self.filename)
def test_mesh_open(self):
import Danu
mesh_name = 'Test Mesh DNE'
self.failUnlessRaises(RuntimeError, self.fh.get_mesh, mesh_name)
Mesh(self.fh, mesh_name, Danu.UNSTRUCTURED_MESH, Danu.LINE_ELEM)
mesh = self.fh.get_mesh(mesh_name)
def test_mesh_destory(self):
import Danu
mesh_name = 'Test Mesh'
mesh = Mesh(self.fh, mesh_name, Danu.UNSTRUCTURED_MESH, Danu.LINE_ELEM)
del mesh
def test_mesh_create_1D(self):
import random
import string
for mesh_type in self.valid_mesh_types:
for elem in self.valid_mesh_elems1:
mesh_name = ''.join(
random.choice(string.letters) for i in xrange(16))
mesh = Mesh(self.fh, mesh_name, mesh_type, elem)
self.assertEqual(mesh.dim, 1, 'Incorrect dimension')
self.assertEqual(mesh.mesh_type, mesh_type,
'Incorrect mesh type')
self.assertEqual(mesh.elem_type, elem,
'Incorrect mesh element type')
def test_mesh_create_2D(self):
import random
import string
for mesh_type in self.valid_mesh_types:
for elem in self.valid_mesh_elems2:
mesh_name = ''.join(
random.choice(string.letters) for i in xrange(16))
mesh = Mesh(self.fh, mesh_name, mesh_type, elem)
self.assertEqual(mesh.dim, 2, 'Incorrect dimension')
self.assertEqual(mesh.mesh_type, mesh_type,
'Incorrect mesh type')
self.assertEqual(mesh.elem_type, elem,
'Incorrect mesh element type')
def test_mesh_create_3D(self):
import random
import string
for mesh_type in self.valid_mesh_types:
for elem in self.valid_mesh_elems3:
mesh_name = ''.join(
random.choice(string.letters) for i in xrange(16))
mesh = Mesh(self.fh, mesh_name, mesh_type, elem)
self.assertEqual(mesh.dim, 3, 'Incorrect dimension')
self.assertEqual(mesh.mesh_type, mesh_type,
'Incorrect mesh type')
self.assertEqual(mesh.elem_type, elem,
'Incorrect mesh element type')
