def setUp(self): """ Setup for test. """ TestQuad4.setUp(self) self.mesh['nvertices'] = 81+18 self.nverticesO = 81 self.fault1Mesh = {'nvertices': 9, 'spaceDim': 3, 'ncells': 8, 'ncorners': 2} self.fault2Mesh = {'nvertices': 9, 'spaceDim': 3, 'ncells': 8, 'ncorners': 2} run_pylith() self.outputRoot = "dislocation2" if has_vtk(): self.reader = VTKDataReader() self.soln = AnalyticalSoln() else: self.reader = None return
def setUp(self): """ Setup for test. """ TestQuad4.setUp(self) self.mesh['nvertices'] = 81 + 9 self.nverticesO = 81 self.faultMesh = { 'nvertices': 9, 'spaceDim': 2, 'ncells': 8, 'ncorners': 2 } run_pylith(LocalApp) self.outputRoot = "dislocation" self.soln = AnalyticalSoln() return
def setUp(self): """ Setup for test. """ TestQuad4.setUp(self) self.mesh['nvertices'] = 81+9 self.nverticesO = 81 self.faultMesh = {'nvertices': 9, 'spaceDim': 2, 'ncells': 8, 'ncorners': 2} run_pylith() self.outputRoot = "dislocation" self.soln = AnalyticalSoln() return
class TestDislocation(TestQuad4): """ Test suite for fault with prescribed slip. """ def setUp(self): """ Setup for test. """ TestQuad4.setUp(self) self.mesh['nvertices'] = 81+9 self.nverticesO = 81 self.faultMesh = {'nvertices': 9, 'spaceDim': 2, 'ncells': 8, 'ncorners': 2} run_pylith() self.outputRoot = "dislocation" self.soln = AnalyticalSoln() return def test_fault_info(self): """ Check fault information. """ if not self.checkResults: return filename = "%s-fault_info.h5" % self.outputRoot fields = ["normal_dir", "final_slip", "slip_time"] check_vertex_fields(self, filename, self.faultMesh, fields) return def test_fault_data(self): """ Check fault information. """ if not self.checkResults: return filename = "%s-fault.h5" % self.outputRoot fields = ["slip", "traction_change"] check_vertex_fields(self, filename, self.faultMesh, fields) return def calcDisplacements(self, vertices): """ Calculate displacement field given coordinates of vertices. """ return self.soln.displacement(vertices, self.nverticesO) def calcStateVar(self, name, vertices, cells): """ Calculate state variable. """ ncells = self.mesh['ncells'] pts = numpy.zeros( (ncells, 3), dtype=numpy.float64) if name == "total_strain": stateVar = self.soln.strain(pts) elif name == "stress": stateVar = self.soln.stress(pts) else: raise ValueError("Unknown state variable '%s'." % name) return stateVar def calcFaultField(self, name, vertices): """ Calculate fault info. """ normalDir = (-1.0, 0.0) finalSlip = -2.0 slipTime = 0.0 nvertices = self.faultMesh['nvertices'] if name == "normal_dir": field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64) field[0,:,0] = normalDir[0] field[0,:,1] = normalDir[1] elif name == "final_slip": field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64) field[0,:,0] = finalSlip elif name == "slip_time": field = slipTime*numpy.zeros( (1, nvertices, 1), dtype=numpy.float64) elif name == "slip": field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64) field[0,:,0] = finalSlip elif name == "traction_change": field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64) field[0,:,0] = 0.0 else: raise ValueError("Unknown fault field '%s'." % name) return field
class TestDislocation2(TestQuad4): """ Test suite for testing pylith with 2-D axial extension. """ def setUp(self): """ Setup for test. """ TestQuad4.setUp(self) self.mesh['nvertices'] = 81+18 self.nverticesO = 81 self.fault1Mesh = {'nvertices': 9, 'spaceDim': 3, 'ncells': 8, 'ncorners': 2} self.fault2Mesh = {'nvertices': 9, 'spaceDim': 3, 'ncells': 8, 'ncorners': 2} run_pylith() self.outputRoot = "dislocation2" if has_vtk(): self.reader = VTKDataReader() self.soln = AnalyticalSoln() else: self.reader = None return def test_fault1_info(self): """ Check fault information. """ if self.reader is None: return filename = "%s-fault1_info.vtk" % self.outputRoot fields = ["normal_dir", "final_slip", "slip_time"] check_vertex_fields(self, filename, self.fault1Mesh, fields) return def test_fault1_data(self): """ Check fault information. """ if self.reader is None: return filename = "%s-fault1_t0000000.vtk" % self.outputRoot fields = ["cumulative_slip", "traction_change"] check_vertex_fields(self, filename, self.fault1Mesh, fields) return def test_fault2_info(self): """ Check fault information. """ if self.reader is None: return filename = "%s-fault2_info.vtk" % self.outputRoot fields = ["normal_dir", "final_slip", "slip_time"] check_vertex_fields(self, filename, self.fault2Mesh, fields) return def test_fault2_data(self): """ Check fault information. """ if self.reader is None: return filename = "%s-fault2_t0000000.vtk" % self.outputRoot fields = ["cumulative_slip", "traction_change"] check_vertex_fields(self, filename, self.fault2Mesh, fields) return def calcDisplacements(self, vertices): """ Calculate displacement field given coordinates of vertices. """ return self.soln.displacement(vertices, self.nverticesO) def calcStateVar(self, name, vertices, cells): """ Calculate state variable. """ ncells = self.mesh['ncells'] pts = numpy.zeros( (ncells, 3), dtype=numpy.float64) if name == "total_strain": stateVar = self.soln.strain(pts) elif name == "stress": stateVar = self.soln.stress(pts) else: raise ValueError("Unknown state variable '%s'." % name) return stateVar def calcFaultField(self, name, vertices): """ Calculate fault info. """ normalDir = (-1.0, 0.0) finalSlip = -2.0 slipTime = 0.0 nvertices = self.faultMesh['nvertices'] if name == "normal_dir": field = numpy.zeros( (nvertices, 3), dtype=numpy.float64) field[:,0] = normalDir[0] field[:,1] = normalDir[1] elif name == "final_slip": field = numpy.zeros( (nvertices, 3), dtype=numpy.float64) field[:,0] = finalSlip elif name == "slip_time": field = slipTime*numpy.zeros( (nvertices, 1), dtype=numpy.float64) elif name == "cumulative_slip": field = numpy.zeros( (nvertices, 3), dtype=numpy.float64) field[:,0] = finalSlip elif name == "traction_change": field = numpy.zeros( (nvertices, 3), dtype=numpy.float64) field[:,0] = 0.0 else: raise ValueError("Unknown fault field '%s'." % name) return field
class TestDislocation(TestQuad4): """ Test suite for fault with prescribed slip. """ def setUp(self): """ Setup for test. """ TestQuad4.setUp(self) self.mesh['nvertices'] = 81 + 9 self.nverticesO = 81 self.faultMesh = { 'nvertices': 9, 'spaceDim': 2, 'ncells': 8, 'ncorners': 2 } run_pylith(LocalApp) self.outputRoot = "dislocation" self.soln = AnalyticalSoln() return def test_fault_info(self): """ Check fault information. """ if not self.checkResults: return filename = "%s-fault_info.h5" % self.outputRoot fields = ["normal_dir", "final_slip", "slip_time"] check_vertex_fields(self, filename, self.faultMesh, fields) return def test_fault_data(self): """ Check fault information. """ if not self.checkResults: return filename = "%s-fault.h5" % self.outputRoot fields = ["slip", "traction_change"] check_vertex_fields(self, filename, self.faultMesh, fields) return def calcDisplacements(self, vertices): """ Calculate displacement field given coordinates of vertices. """ return self.soln.displacement(vertices, self.nverticesO) def calcStateVar(self, name, vertices, cells): """ Calculate state variable. """ ncells = self.mesh['ncells'] pts = numpy.zeros((ncells, 3), dtype=numpy.float64) if name == "total_strain": stateVar = self.soln.strain(pts) elif name == "stress": stateVar = self.soln.stress(pts) else: raise ValueError("Unknown state variable '%s'." % name) return stateVar def calcFaultField(self, name, vertices): """ Calculate fault info. """ normalDir = (-1.0, 0.0) finalSlip = -2.0 slipTime = 0.0 nvertices = self.faultMesh['nvertices'] if name == "normal_dir": field = numpy.zeros((1, nvertices, 2), dtype=numpy.float64) field[0, :, 0] = normalDir[0] field[0, :, 1] = normalDir[1] elif name == "final_slip": field = numpy.zeros((1, nvertices, 2), dtype=numpy.float64) field[0, :, 0] = finalSlip elif name == "slip_time": field = slipTime * numpy.zeros( (1, nvertices, 1), dtype=numpy.float64) elif name == "slip": field = numpy.zeros((1, nvertices, 2), dtype=numpy.float64) field[0, :, 0] = finalSlip elif name == "traction_change": field = numpy.zeros((1, nvertices, 2), dtype=numpy.float64) field[0, :, 0] = 0.0 else: raise ValueError("Unknown fault field '%s'." % name) return field