def test_read_velocity_model(self):
for filename in ['iasp91.tvel', 'iasp91_w_comment.tvel', 'iasp91.nd',
'iasp91_w_comment.nd']:
velocity_model = os.path.join(DATA, filename)
test2 = VelocityModel.readVelocityFile(velocity_model)
self.assertEqual(len(test2.layers), 129)
self.assertEqual(len(test2), 129)
self.assertEqual(test2.radiusOfEarth, 6371.0)
self.assertEqual(test2.mohoDepth, 35)
self.assertEqual(test2.cmbDepth, 2889.0)
self.assertEqual(test2.iocbDepth, 5153.9)
self.assertEqual(test2.minRadius, 0.0)
self.assertEqual(test2.maxRadius, 6371.0)
self.assertEqual(test2.validate(), True)
np.testing.assert_equal(
test2.getDisconDepths(),
[0.0, 20.0, 35.0, 210.0, 410.0, 660.0, 2889.0, 5153.9, 6371.0])
# check boundary cases
self.assertEqual(test2.layerNumberAbove(6371), 128)
self.assertEqual(test2.layerNumberBelow(0), 0)
# evaluate at cmb
self.assertEqual(test2.evaluateAbove(2889.0, 'p'), 13.6908)
self.assertEqual(test2.evaluateBelow(2889.0, 'D'), 9.9145)
self.assertEqual(test2.depthAtTop(50), 2393.5)
self.assertEqual(test2.depthAtBottom(50), 2443.0)
self.assertEqual(test2.fixDisconDepths(), False)
def test_read_velocity_model(self):
for filename in [
'iasp91.tvel', 'iasp91_w_comment.tvel', 'iasp91.nd',
'iasp91_w_comment.nd'
]:
velocity_model = os.path.join(DATA, filename)
test2 = VelocityModel.readVelocityFile(velocity_model)
self.assertEqual(len(test2.layers), 129)
self.assertEqual(len(test2), 129)
self.assertEqual(test2.radiusOfEarth, 6371.0)
self.assertEqual(test2.mohoDepth, 35)
self.assertEqual(test2.cmbDepth, 2889.0)
self.assertEqual(test2.iocbDepth, 5153.9)
self.assertEqual(test2.minRadius, 0.0)
self.assertEqual(test2.maxRadius, 6371.0)
self.assertEqual(test2.validate(), True)
np.testing.assert_equal(
test2.getDisconDepths(),
[0.0, 20.0, 35.0, 210.0, 410.0, 660.0, 2889.0, 5153.9, 6371.0])
# check boundary cases
self.assertEqual(test2.layerNumberAbove(6371), 128)
self.assertEqual(test2.layerNumberBelow(0), 0)
# evaluate at cmb
self.assertEqual(test2.evaluateAbove(2889.0, 'p'), 13.6908)
self.assertEqual(test2.evaluateBelow(2889.0, 'D'), 9.9145)
self.assertEqual(test2.depthAtTop(50), 2393.5)
self.assertEqual(test2.depthAtBottom(50), 2443.0)
self.assertEqual(test2.fixDisconDepths(), False)
def loadVMod(self):
"""
Try to load a velocity model.
"""
# Read the velocity model file.
filename = self.input_filename
if self.debug:
print("filename =", filename)
self.vMod = VelocityModel.readVelocityFile(filename)
if self.vMod is None:
raise IOError("Velocity model file not found: " + filename)
# If model was read:
if self.debug:
print("Done reading velocity model.")
print("Radius of model " + self.vMod.modelName + " is " +
str(self.vMod.radiusOfEarth))
# if self.debug:
# print("velocity mode: " + self.vMod)
return self.vMod
def loadVMod(self):
"""
Try to load a velocity model.
"""
# Read the velocity model file.
filename = self.input_filename
if self.debug:
print("filename =", filename)
self.vMod = VelocityModel.readVelocityFile(filename)
if self.vMod is None:
raise IOError("Velocity model file not found: " + filename)
# If model was read:
if self.debug:
print("Done reading velocity model.")
print("Radius of model " + self.vMod.modelName + " is " +
str(self.vMod.radiusOfEarth))
# if self.debug:
# print("velocity mode: " + self.vMod)
return self.vMod
