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 = "friction_shear_stick"
self.soln = AnalyticalSoln()
return
class TestFrictionShearStick(TestQuad4):
"""
Test suite for testing pylith with 2-D shear stick with friction.
"""
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 = "friction_shear_stick"
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 = ["strike_dir", "normal_dir", "traction_initial","friction_coefficient","cohesion"]
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"]
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.
"""
strikeDir = (0.0, -1.0)
normalDir = (-1.0, 0.0)
initialTraction = (0.0, -1.0e+7)
frictionCoefficient = 0.6
nvertices = self.faultMesh['nvertices']
if name == "strike_dir":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = strikeDir[0]
field[0,:,1] = strikeDir[1]
elif name == "normal_dir":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = normalDir[0]
field[0,:,1] = normalDir[1]
elif name == "traction_initial":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = initialTraction[0]
field[0,:,1] = initialTraction[1]
elif name == "friction_coefficient":
field = numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
field[0,:,0] = frictionCoefficient
elif name == "cohesion":
field = numpy.zeros( (1, nvertices, 1), dtype=numpy.float64)
elif name == "slip":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
elif name == "traction":
field = numpy.zeros( (1, nvertices, 2), dtype=numpy.float64)
field[0,:,0] = 1.0e+6
field[0,:,1] = -1.0e+7
else:
raise ValueError("Unknown fault field '%s'." % name)
return field