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_checkWrite(self):
"""
Test _checkWrite().
"""
dataProvider = TestProvider()
# Default values should be true
output = OutputManager()
output.inventory.writer._configure()
output._configure()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
self.assertEqual(True, output._checkWrite(0.0))
self.assertEqual(True, output._checkWrite(3.234e+8))
# Check writing based on time
output = OutputManager()
output.inventory.writer._configure()
output._configure()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
output.inventory.outputFreq = "time_step"
t = 0.0
dt = output.dtN
self.assertEqual(True, output._checkWrite(t))
self.assertEqual(False, output._checkWrite(t))
self.assertEqual(False, output._checkWrite(t + 0.8 * dt))
t += dt
self.assertEqual(True, output._checkWrite(t))
t = 2 * dt
self.assertEqual(True, output._checkWrite(t))
# Check writing based on number of steps
output = OutputManager()
output.inventory.writer._configure()
output.inventory.outputFreq = "skip"
output.inventory.skip = 1
output._configure()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
t = 0.0
dt = 1.0
self.assertEqual(True, output._checkWrite(t))
t += dt
self.assertEqual(False, output._checkWrite(t))
t += dt
self.assertEqual(True, output._checkWrite(t))
output.inventory.skip = 2
t += dt
self.assertEqual(False, output._checkWrite(t))
t += dt
self.assertEqual(False, output._checkWrite(t))
t += dt
self.assertEqual(True, output._checkWrite(t))
return
def test_constructor(self): """Test constructor. """ output = OutputManager() output.inventory.writer._configure() output._configure() return
def test_preinitialize(self): """Test preinitialize(). """ dataProvider = TestProvider() output = OutputManager() output.preinitialize(dataProvider) self.failIf(output.dataProvider is None) return
def test_estimateNumSteps(self):
"""
Test _estimateNumSteps().
"""
from pyre.units.time import second
output = OutputManager()
output.inventory.outputFreq = "skip"
output.inventory.skip = 2
output._configure()
numTimeSteps = 0
totalTime = 1.0 * second
self.assertEqual(0, output._estimateNumSteps(totalTime, numTimeSteps))
numTimeSteps = 4
totalTime = 1.0 * second
self.assertEqual(2, output._estimateNumSteps(totalTime, numTimeSteps))
numTimeSteps = 2
totalTime = 1.0 * second
self.assertEqual(1, output._estimateNumSteps(totalTime, numTimeSteps))
output = OutputManager()
output.inventory.outputFreq = "time_step"
output.inventory.dt = 2.0 * second
output._configure()
dataProvider = TestProvider()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
numTimeSteps = 0
totalTime = 1.0 * second
self.assertEqual(0, output._estimateNumSteps(totalTime, numTimeSteps))
numTimeSteps = 4
totalTime = 1.0 * second
self.assertEqual(1, output._estimateNumSteps(totalTime, numTimeSteps))
numTimeSteps = 2
totalTime = 2.0 * second
self.assertEqual(2, output._estimateNumSteps(totalTime, numTimeSteps))
return
def test_verifyConfiguration(self):
"""Test verifyConfiguration().
"""
dataProvider = TestProvider()
output = OutputManager()
output.preinitialize(dataProvider)
output.vertexInfoFields = ["vertex info"]
output.vertexDataFields = ["vertex data 2"]
output.cellInfoFields = []
output.cellDataFields = ["cell data"]
output.verifyConfiguration(dataProvider.getDataMesh())
return
def test_openclose(self):
"""Test open() and close().
"""
output = OutputManager()
output.inventory.writer.inventory.filename = "output.vtk"
output.inventory.writer._configure()
output._configure()
dataProvider = TestProvider()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
output.open(totalTime=5.0, numTimeSteps=2)
output.close()
return
def test_writeInfo(self): """Test writeInfo(). """ output = OutputManager() output.inventory.writer.inventory.filename = "output.vtk" output.inventory.writer._configure() output.inventory.vertexInfoFields = ["vertex info"] output.inventory.cellInfoFields = ["cell info"] output._configure() dataProvider = TestProvider() output.preinitialize(dataProvider) output.initialize(self.normalizer) output.open(totalTime=5.0, numTimeSteps=2) output.writeInfo() output.close() return
def test_writeData(self):
"""Test writeData().
"""
output = OutputManager()
output.inventory.writer.inventory.filename = "output.vtk"
output.inventory.writer.inventory.timeFormat = "%3.1f"
output.inventory.writer._configure()
output.inventory.vertexDataFields = ["vertex data 2", "vertex data 1"]
output.inventory.cellDataFields = ["cell data"]
output._configure()
dataProvider = TestProvider()
output.preinitialize(dataProvider)
output.initialize(self.normalizer)
output.open(totalTime=5.0, numTimeSteps=2)
output.writeData(2.0, dataProvider.fields)
output.close()
return
