def test_initialize(self):
"""Test initialize().
"""
# No quadrature
output = OutputManager()
output.inventory.writer.inventory.filename = "test.vtk"
output.inventory.writer._configure()
output._configure()
dataProvider = TestProvider()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
# With quadrature
from pylith.feassemble.FIATLagrange import FIATLagrange
from pylith.feassemble.Quadrature import Quadrature
cell = FIATLagrange()
cell.inventory.dimension = 2
cell.inventory.degree = 2
cell.inventory.order = 2
cell._configure()
quadrature = Quadrature()
quadrature.inventory.cell = cell
quadrature._configure()
output = OutputManager()
output.inventory.writer.inventory.filename = "test.vtk"
output.inventory.writer._configure()
output._configure()
dataProvider = TestProvider()
output.preinitialize(dataProvider)
output.initialize(self.normalizer, quadrature)
return
def test_initialize_hex27(self):
"""
Test initialize() with hex27 cell.
"""
cell = FIATLagrange()
cell.inventory.dimension = 3
cell.inventory.degree = 2
cell.inventory.order = 3
cell._configure()
cell.initialize(spaceDim=3)
cellE = Hex27()
self._checkVals(cellE, cell)
from pylith.feassemble.CellGeometry import GeometryHex3D
self.failUnless(isinstance(cell.geometry, GeometryHex3D))
return
def test_initialize_quad4(self):
"""
Test initialize() with quad4 cell.
"""
cell = FIATLagrange()
cell.inventory.dimension = 2
cell.inventory.degree = 1
cell.inventory.order = 2
cell._configure()
cell.initialize(spaceDim=2)
cellE = Quad4()
self._checkVals(cellE, cell)
from pylith.feassemble.CellGeometry import GeometryQuad2D
self.failUnless(isinstance(cell.geometry, GeometryQuad2D))
return
def test_lagrange3D(self):
"""
Test setup of quadrature for Lagrange cells for a 3-D problem.
"""
spaceDim = 3
cell = FIATLagrange()
cell.inventory.dimension = 3
cell._configure()
quadrature = Quadrature()
quadrature.inventory.cell = cell
quadrature._configure()
quadrature.preinitialize(spaceDim)
self.assertEqual(3, quadrature.cellDim())
self.assertEqual(spaceDim, quadrature.spaceDim())
self.assertEqual(8, quadrature.numBasis())
return
