def test_readQuakeML(self):
"""
"""
# IRIS
filename = os.path.join(self.path, 'iris_events.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 2)
self.assertEqual(
catalog[0].resource_id,
ResourceIdentifier(
'smi:www.iris.edu/ws/event/query?eventId=3279407'))
self.assertEqual(
catalog[1].resource_id,
ResourceIdentifier(
'smi:www.iris.edu/ws/event/query?eventId=2318174'))
# NERIES
filename = os.path.join(self.path, 'neries_events.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 3)
self.assertEqual(
catalog[0].resource_id,
ResourceIdentifier('quakeml:eu.emsc/event/20120404_0000041'))
self.assertEqual(
catalog[1].resource_id,
ResourceIdentifier('quakeml:eu.emsc/event/20120404_0000038'))
self.assertEqual(
catalog[2].resource_id,
ResourceIdentifier('quakeml:eu.emsc/event/20120404_0000039'))
def readSeisComPEventXML0_6(filename):
"""
Reads a single SeisComP event XML V0.6 file and returns a ObsPy Catalog object.
This fixes the following differences to QUAKEML1.1:
- EventProperties is replaced by eventProperties
- event start tag is moved behind eventProperties start tag
- magnitude nodes are moved out of origin nodes
- an origin_id tag is inserted into every magnitude node
- the magnitude tag giving the value and uncertainty of the magnitude is
replaced by a mag tag
"""
# realy dirty hack
spans = [0]
xml = open(filename, 'rt').read()
m_event = ev_expr.search(xml)
ins = '<eventParameters>' + m_event.group()
for m in mag_expr.finditer(xml):
ins += m.expand(r'\1\2\3<originID>\2</originID>\4mag\6mag\8')
spans.extend(list(m.span()))
spans.extend(list(m_event.span()) + [-1])
xml = ''.join(
[xml[spans[2 * i]:spans[2 * i + 1]] for i in range(len(spans) // 2)])
xml = ev_para_expr.sub(ins, xml)
xml = xml.replace('EventParameters', 'eventParameters')
temp = StringIO.StringIO(xml)
return readQuakeML(temp)
def test_reading_twice_raises_no_warning(self):
"""
Tests that reading a QuakeML file twice does not raise a warnings.
Not an extensive test but likely good enough.
"""
filename = os.path.join(self.path, "qml-example-1.2-RC3.xml")
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
cat1 = readQuakeML(filename)
self.assertEqual(len(w), 0)
cat2 = readQuakeML(filename)
self.assertEqual(len(w), 0)
self.assertEqual(cat1, cat2)
def test_stationmagnitude(self):
"""
Tests StationMagnitude object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_stationmagnitude.xml')
catalog = readQuakeML(filename)
self.assertEquals(len(catalog), 1)
self.assertEquals(len(catalog[0].station_magnitudes), 1)
mag = catalog[0].station_magnitudes[0]
# Assert the actual StationMagnitude object. Everything that is not set
# in the QuakeML file should be set to None.
self.assertEqual(
mag.resource_id,
ResourceIdentifier("smi:ch.ethz.sed/magnitude/station/881342"))
self.assertEquals(mag.origin_id,
ResourceIdentifier('smi:some/example/id'))
self.assertEquals(mag.mag, 6.5)
self.assertEquals(mag.mag_errors.uncertainty, 0.2)
self.assertEquals(mag.station_magnitude_type, 'MS')
self.assertEqual(
mag.amplitude_id,
ResourceIdentifier("smi:ch.ethz.sed/amplitude/824315"))
self.assertEqual(mag.method_id,
ResourceIdentifier(\
"smi:ch.ethz.sed/magnitude/generic/surface_wave_magnitude"))
self.assertEqual(
mag.waveform_id,
WaveformStreamID(network_code='BW',
station_code='FUR',
resource_uri="smi:ch.ethz.sed/waveform/201754"))
self.assertEqual(mag.creation_info, None)
# exporting back to XML should result in the same document
original = open(filename, "rt").read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def test_write_amplitude_time_window(self):
"""
Tests writing an QuakeML Amplitude with TimeWindow.
"""
filename = os.path.join(self.path, "qml-example-1.2-RC3.xml")
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
cat = readQuakeML(filename)
self.assertEqual(len(w), 0)
with NamedTemporaryFile() as tf:
tmpfile = tf.name
cat.write(tmpfile, format='QUAKEML')
with open(tmpfile, "rb") as fh:
lines = fh.readlines()
firstline = 45
while b"<amplitude " not in lines[firstline]:
firstline += 1
got = [
lines[i_].strip() for i_ in range(firstline, firstline + 13)
]
expected = [
b'<amplitude publicID="smi:nz.org.geonet/event/2806038g/'
b'amplitude/1/modified">', b'<genericAmplitude>',
b'<value>1e-08</value>', b'</genericAmplitude>',
b'<type>A</type>', b'<category>point</category>',
b'<unit>m/s</unit>', b'<timeWindow>',
b'<reference>2007-10-10T14:40:39.055000Z</reference>',
b'<begin>0.0</begin>', b'<end>0.51424</end>', b'</timeWindow>',
b'</amplitude>'
]
self.assertEqual(got, expected)
def test_writeQuakeML(self):
"""
Tests writing a QuakeML document.
"""
filename = os.path.join(self.path, 'qml-example-1.2-RC3.xml')
with NamedTemporaryFile() as tf:
tmpfile = tf.name
catalog = readQuakeML(filename)
self.assertTrue(len(catalog), 1)
writeQuakeML(catalog, tmpfile, validate=IS_RECENT_LXML)
# Read file again. Avoid the (legit) warning about the already used
# resource identifiers.
with warnings.catch_warnings(record=True):
warnings.simplefilter("ignore")
catalog2 = readQuakeML(tmpfile)
self.assertTrue(len(catalog2), 1)
def test_arrival(self):
"""
Tests Arrival object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_arrival.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
self.assertEqual(len(catalog[0].origins[0].arrivals), 2)
ar = catalog[0].origins[0].arrivals[0]
# Test the actual Arrival object. Everything not set in the QuakeML
# file should be None.
self.assertEqual(ar.pick_id,
ResourceIdentifier('smi:ch.ethz.sed/pick/117634'))
self.assertEqual(ar.phase, 'Pn')
self.assertEqual(ar.azimuth, 12.0)
self.assertEqual(ar.distance, 0.5)
self.assertEqual(ar.takeoff_angle, 11.0)
self.assertEqual(ar.takeoff_angle_errors.uncertainty, 0.2)
self.assertEqual(ar.time_residual, 1.6)
self.assertEqual(ar.horizontal_slowness_residual, 1.7)
self.assertEqual(ar.backazimuth_residual, 1.8)
self.assertEqual(ar.time_weight, 0.48)
self.assertEqual(ar.horizontal_slowness_weight, 0.49)
self.assertEqual(ar.backazimuth_weight, 0.5)
self.assertEqual(ar.earth_model_id,
ResourceIdentifier('smi:ch.ethz.sed/earthmodel/U21'))
self.assertEqual(len(ar.comments), 1)
self.assertEqual(ar.creation_info.author, "Erika Mustermann")
# exporting back to XML should result in the same document
original = open(filename, "rt").read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def test_stationmagnitudecontribution(self):
"""
Tests the station magnitude contribution object.
"""
filename = os.path.join(
self.path, 'quakeml_1.2_stationmagnitudecontributions.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
self.assertEqual(len(catalog[0].magnitudes), 1)
self.assertEqual(
len(catalog[0].magnitudes[0].station_magnitude_contributions), 2)
# Check the first stationMagnitudeContribution object.
stat_contrib = \
catalog[0].magnitudes[0].station_magnitude_contributions[0]
self.assertEqual(stat_contrib.station_magnitude_id.resource_id,
"smi:ch.ethz.sed/magnitude/station/881342")
self.assertEqual(stat_contrib.weight, 0.77)
self.assertEqual(stat_contrib.residual, 0.02)
# Check the second stationMagnitudeContribution object.
stat_contrib = \
catalog[0].magnitudes[0].station_magnitude_contributions[1]
self.assertEqual(stat_contrib.station_magnitude_id.resource_id,
"smi:ch.ethz.sed/magnitude/station/881334")
self.assertEqual(stat_contrib.weight, 0.55)
self.assertEqual(stat_contrib.residual, 0.11)
# exporting back to XML should result in the same document
original = open(filename, "rt").read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def test_writeQuakeML(self):
"""
Tests writing a QuakeML document.
skipIfPython25 due to the use of the warnings context manager.
"""
filename = os.path.join(self.path, 'qml-example-1.2-RC3.xml')
tmpfile = NamedTemporaryFile().name
catalog = readQuakeML(filename)
self.assertTrue(len(catalog), 1)
writeQuakeML(catalog, tmpfile)
# Read file again. Avoid the (legit) warning about the already used
# resource identifiers.
with warnings.catch_warnings(record=True):
warnings.simplefilter("ignore")
catalog2 = readQuakeML(tmpfile)
self.assertTrue(len(catalog2), 1)
# clean up
os.remove(tmpfile)
def test_read_equivalence(self):
"""
See #662.
Tests if readQuakeML() and readEvents() return the same results.
"""
warnings.simplefilter("ignore", UserWarning)
cat1 = readEvents(self.neries_filename)
cat2 = readQuakeML(self.neries_filename)
warnings.filters.pop(0)
self.assertEqual(cat1, cat2)
def test_event(self):
"""
Tests Event object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_event.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
event = catalog[0]
self.assertEqual(
event.resource_id,
ResourceIdentifier('smi:ch.ethz.sed/event/historical/1165'))
# enums
self.assertEqual(event.event_type, 'earthquake')
self.assertEqual(event.event_type_certainty, 'suspected')
# comments
self.assertEqual(len(event.comments), 2)
c = event.comments
self.assertEqual(c[0].text, 'Relocated after re-evaluation')
self.assertEqual(c[0].resource_id, None)
self.assertEqual(c[0].creation_info.agency_id, 'EMSC')
self.assertEqual(c[1].text, 'Another comment')
self.assertEqual(
c[1].resource_id,
ResourceIdentifier(id="smi:some/comment/id/number_3"))
self.assertEqual(c[1].creation_info, None)
# event descriptions
self.assertEqual(len(event.event_descriptions), 3)
d = event.event_descriptions
self.assertEqual(d[0].text, '1906 San Francisco Earthquake')
self.assertEqual(d[0].type, 'earthquake name')
self.assertEqual(d[1].text, 'NEAR EAST COAST OF HONSHU, JAPAN')
self.assertEqual(d[1].type, 'Flinn-Engdahl region')
self.assertEqual(d[2].text, 'free-form string')
self.assertEqual(d[2].type, None)
# creation info
self.assertEqual(event.creation_info.author, "Erika Mustermann")
self.assertEqual(event.creation_info.agency_id, "EMSC")
self.assertEqual(
event.creation_info.author_uri,
ResourceIdentifier("smi:smi-registry/organization/EMSC"))
self.assertEqual(
event.creation_info.agency_uri,
ResourceIdentifier("smi:smi-registry/organization/EMSC"))
self.assertEqual(
event.creation_info.creation_time,
UTCDateTime("2012-04-04T16:40:50+00:00"))
self.assertEqual(event.creation_info.version, "1.0.1")
# exporting back to XML should result in the same document
with open(filename, "rt") as fp:
original = fp.read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def test_magnitude(self):
"""
Tests Magnitude object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_magnitude.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
self.assertEqual(len(catalog[0].magnitudes), 1)
mag = catalog[0].magnitudes[0]
self.assertEqual(
mag.resource_id,
ResourceIdentifier('smi:ch.ethz.sed/magnitude/37465'))
self.assertEqual(mag.mag, 5.5)
self.assertEqual(mag.mag_errors.uncertainty, 0.1)
self.assertEqual(mag.magnitude_type, 'MS')
self.assertEqual(
mag.method_id,
ResourceIdentifier(
'smi:ch.ethz.sed/magnitude/generic/surface_wave_magnitude'))
self.assertEqual(mag.station_count, 8)
self.assertEqual(mag.evaluation_status, 'preliminary')
# comments
self.assertEqual(len(mag.comments), 2)
c = mag.comments
self.assertEqual(c[0].text, 'Some comment')
self.assertEqual(
c[0].resource_id,
ResourceIdentifier(id="smi:some/comment/id/muh"))
self.assertEqual(c[0].creation_info.author, 'EMSC')
self.assertEqual(c[1].creation_info, None)
self.assertEqual(c[1].text, 'Another comment')
self.assertEqual(c[1].resource_id, None)
# creation info
self.assertEqual(mag.creation_info.author, "NEIC")
self.assertEqual(mag.creation_info.agency_id, None)
self.assertEqual(mag.creation_info.author_uri, None)
self.assertEqual(mag.creation_info.agency_uri, None)
self.assertEqual(mag.creation_info.creation_time, None)
self.assertEqual(mag.creation_info.version, None)
# exporting back to XML should result in the same document
with open(filename, "rt") as fp:
original = fp.read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def test_data_used_in_moment_tensor(self):
"""
Tests the data used objects in moment tensors.
"""
filename = os.path.join(self.path, 'quakeml_1.2_data_used.xml')
# Test reading first.
catalog = readQuakeML(filename)
event = catalog[0]
self.assertTrue(len(event.focal_mechanisms), 2)
# First focmec contains only one data used element.
self.assertEqual(
len(event.focal_mechanisms[0].moment_tensor.data_used), 1)
du = event.focal_mechanisms[0].moment_tensor.data_used[0]
self.assertEqual(du.wave_type, "body waves")
self.assertEqual(du.station_count, 88)
self.assertEqual(du.component_count, 166)
self.assertEqual(du.shortest_period, 40.0)
# Second contains three. focmec contains only one data used element.
self.assertEqual(
len(event.focal_mechanisms[1].moment_tensor.data_used), 3)
du = event.focal_mechanisms[1].moment_tensor.data_used
self.assertEqual(du[0].wave_type, "body waves")
self.assertEqual(du[0].station_count, 88)
self.assertEqual(du[0].component_count, 166)
self.assertEqual(du[0].shortest_period, 40.0)
self.assertEqual(du[1].wave_type, "surface waves")
self.assertEqual(du[1].station_count, 96)
self.assertEqual(du[1].component_count, 189)
self.assertEqual(du[1].shortest_period, 50.0)
self.assertEqual(du[2].wave_type, "mantle waves")
self.assertEqual(du[2].station_count, 41)
self.assertEqual(du[2].component_count, 52)
self.assertEqual(du[2].shortest_period, 125.0)
# exporting back to XML should result in the same document
with open(filename, "rt") as fp:
original = fp.read()
processed = Pickler().dumps(catalog)
compare_xml_strings(original, processed)
def test_read_amplitude_time_window(self):
"""
Tests reading an QuakeML Amplitude with TimeWindow.
"""
filename = os.path.join(self.path, "qml-example-1.2-RC3.xml")
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
cat = readQuakeML(filename)
self.assertEqual(len(w), 0)
self.assertEqual(len(cat[0].amplitudes), 1)
amp = cat[0].amplitudes[0]
self.assertEqual(amp.type, "A")
self.assertEqual(amp.category, "point")
self.assertEqual(amp.unit, "m/s")
self.assertEqual(amp.generic_amplitude, 1e-08)
self.assertEqual(amp.time_window.begin, 0.0)
self.assertEqual(amp.time_window.end, 0.51424)
self.assertEqual(amp.time_window.reference,
UTCDateTime("2007-10-10T14:40:39.055"))
def test_pick(self):
"""
Tests Pick object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_pick.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
self.assertEqual(len(catalog[0].picks), 2)
pick = catalog[0].picks[0]
self.assertEqual(pick.resource_id,
ResourceIdentifier('smi:ch.ethz.sed/pick/117634'))
self.assertEqual(pick.time, UTCDateTime('2005-09-18T22:04:35Z'))
self.assertEqual(pick.time_errors.uncertainty, 0.012)
self.assertEqual(
pick.waveform_id,
WaveformStreamID(network_code='BW',
station_code='FUR',
resource_uri='smi:ch.ethz.sed/waveform/201754'))
self.assertEqual(
pick.filter_id,
ResourceIdentifier('smi:ch.ethz.sed/filter/lowpass/standard'))
self.assertEqual(
pick.method_id,
ResourceIdentifier('smi:ch.ethz.sed/picker/autopicker/6.0.2'))
self.assertEqual(pick.backazimuth, 44.0)
self.assertEqual(pick.onset, 'impulsive')
self.assertEqual(pick.phase_hint, 'Pn')
self.assertEqual(pick.polarity, 'positive')
self.assertEqual(pick.evaluation_mode, "manual")
self.assertEqual(pick.evaluation_status, "confirmed")
self.assertEqual(len(pick.comments), 2)
self.assertEqual(pick.creation_info.author, "Erika Mustermann")
# exporting back to XML should result in the same document
with open(filename, "rt") as fp:
original = fp.read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def test_creating_minimal_QuakeML_with_MT(self):
"""
Tests the creation of a minimal QuakeML containing origin, magnitude
and moment tensor.
"""
# Rotate into physical domain
lat, lon, depth, org_time = 10.0, -20.0, 12000, UTCDateTime(2012, 1, 1)
mrr, mtt, mpp, mtr, mpr, mtp = 1E18, 2E18, 3E18, 3E18, 2E18, 1E18
scalar_moment = math.sqrt(mrr**2 + mtt**2 + mpp**2 + mtr**2 + mpr**2 +
mtp**2)
moment_magnitude = 0.667 * (math.log10(scalar_moment) - 9.1)
# Initialise event
ev = Event(event_type="earthquake")
ev_origin = Origin(time=org_time,
latitude=lat,
longitude=lon,
depth=depth,
resource_id=ResourceIdentifier())
ev.origins.append(ev_origin)
# populte event moment tensor
ev_tensor = Tensor(m_rr=mrr,
m_tt=mtt,
m_pp=mpp,
m_rt=mtr,
m_rp=mpr,
m_tp=mtp)
ev_momenttensor = MomentTensor(tensor=ev_tensor)
ev_momenttensor.scalar_moment = scalar_moment
ev_momenttensor.derived_origin_id = ev_origin.resource_id
ev_focalmechanism = FocalMechanism(moment_tensor=ev_momenttensor)
ev.focal_mechanisms.append(ev_focalmechanism)
# populate event magnitude
ev_magnitude = Magnitude()
ev_magnitude.mag = moment_magnitude
ev_magnitude.magnitude_type = 'Mw'
ev_magnitude.evaluation_mode = 'automatic'
ev.magnitudes.append(ev_magnitude)
# write QuakeML file
cat = Catalog(events=[ev])
memfile = StringIO.StringIO()
cat.write(memfile, format="quakeml", validate=IS_RECENT_LXML)
memfile.seek(0, 0)
new_cat = readQuakeML(memfile)
self.assertEqual(len(new_cat), 1)
event = new_cat[0]
self.assertEqual(len(event.origins), 1)
self.assertEqual(len(event.magnitudes), 1)
self.assertEqual(len(event.focal_mechanisms), 1)
org = event.origins[0]
mag = event.magnitudes[0]
fm = event.focal_mechanisms[0]
self.assertEqual(org.latitude, lat)
self.assertEqual(org.longitude, lon)
self.assertEqual(org.depth, depth)
self.assertEqual(org.time, org_time)
# Moment tensor.
mt = fm.moment_tensor.tensor
self.assertTrue((fm.moment_tensor.scalar_moment - scalar_moment) /
scalar_moment < scalar_moment * 1E-10)
self.assertEqual(mt.m_rr, mrr)
self.assertEqual(mt.m_pp, mpp)
self.assertEqual(mt.m_tt, mtt)
self.assertEqual(mt.m_rt, mtr)
self.assertEqual(mt.m_rp, mpr)
self.assertEqual(mt.m_tp, mtp)
# Mag
self.assertAlmostEqual(mag.mag, moment_magnitude)
self.assertEqual(mag.magnitude_type, "Mw")
self.assertEqual(mag.evaluation_mode, "automatic")
def test_focalmechanism(self):
"""
Tests FocalMechanism object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_focalmechanism.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
self.assertEqual(len(catalog[0].focal_mechanisms), 2)
fm = catalog[0].focal_mechanisms[0]
# general
self.assertEqual(fm.resource_id,
ResourceIdentifier('smi:ISC/fmid=292309'))
self.assertEqual(fm.waveform_id.network_code, 'BW')
self.assertEqual(fm.waveform_id.station_code, 'FUR')
self.assertEqual(
fm.waveform_id.resource_uri,
ResourceIdentifier(resource_id="smi:ch.ethz.sed/waveform/201754"))
self.assertTrue(isinstance(fm.waveform_id, WaveformStreamID))
self.assertEqual(fm.triggering_origin_id,
ResourceIdentifier('smi:local/originId=7680412'))
self.assertAlmostEqual(fm.azimuthal_gap, 0.123)
self.assertEqual(fm.station_polarity_count, 987)
self.assertAlmostEqual(fm.misfit, 1.234)
self.assertAlmostEqual(fm.station_distribution_ratio, 2.345)
self.assertEqual(
fm.method_id,
ResourceIdentifier('smi:ISC/methodID=Best_double_couple'))
# comments
self.assertEqual(len(fm.comments), 2)
c = fm.comments
self.assertEqual(c[0].text, 'Relocated after re-evaluation')
self.assertEqual(c[0].resource_id, None)
self.assertEqual(c[0].creation_info.agency_id, 'MUH')
self.assertEqual(c[1].text, 'Another MUH')
self.assertEqual(
c[1].resource_id,
ResourceIdentifier(resource_id="smi:some/comment/id/number_3"))
self.assertEqual(c[1].creation_info, None)
# creation info
self.assertEqual(fm.creation_info.author, "Erika Mustermann")
self.assertEqual(fm.creation_info.agency_id, "MUH")
self.assertEqual(
fm.creation_info.author_uri,
ResourceIdentifier("smi:smi-registry/organization/MUH"))
self.assertEqual(
fm.creation_info.agency_uri,
ResourceIdentifier("smi:smi-registry/organization/MUH"))
self.assertEqual(fm.creation_info.creation_time,
UTCDateTime("2012-04-04T16:40:50+00:00"))
self.assertEqual(fm.creation_info.version, "1.0.1")
# nodalPlanes
self.assertAlmostEqual(fm.nodal_planes.nodal_plane_1.strike, 346.0)
self.assertAlmostEqual(fm.nodal_planes.nodal_plane_1.dip, 57.0)
self.assertAlmostEqual(fm.nodal_planes.nodal_plane_1.rake, 75.0)
self.assertAlmostEqual(fm.nodal_planes.nodal_plane_2.strike, 193.0)
self.assertAlmostEqual(fm.nodal_planes.nodal_plane_2.dip, 36.0)
self.assertAlmostEqual(fm.nodal_planes.nodal_plane_2.rake, 112.0)
self.assertEqual(fm.nodal_planes.preferred_plane, 2)
# principalAxes
self.assertAlmostEqual(fm.principal_axes.t_axis.azimuth, 216.0)
self.assertAlmostEqual(fm.principal_axes.t_axis.plunge, 73.0)
self.assertAlmostEqual(fm.principal_axes.t_axis.length, 1.050e+18)
self.assertAlmostEqual(fm.principal_axes.p_axis.azimuth, 86.0)
self.assertAlmostEqual(fm.principal_axes.p_axis.plunge, 10.0)
self.assertAlmostEqual(fm.principal_axes.p_axis.length, -1.180e+18)
self.assertEqual(fm.principal_axes.n_axis.azimuth, None)
self.assertEqual(fm.principal_axes.n_axis.plunge, None)
self.assertEqual(fm.principal_axes.n_axis.length, None)
# momentTensor
mt = fm.moment_tensor
self.assertEqual(mt.resource_id,
ResourceIdentifier('smi:ISC/mtid=123321'))
self.assertEqual(mt.derived_origin_id,
ResourceIdentifier('smi:ISC/origid=13145006'))
self.assertAlmostEqual(mt.scalar_moment, 1.100e+18)
self.assertAlmostEqual(mt.tensor.m_rr, 9.300e+17)
self.assertAlmostEqual(mt.tensor.m_tt, 1.700e+17)
self.assertAlmostEqual(mt.tensor.m_pp, -1.100e+18)
self.assertAlmostEqual(mt.tensor.m_rt, -2.200e+17)
self.assertAlmostEqual(mt.tensor.m_rp, 4.000e+17)
self.assertAlmostEqual(mt.tensor.m_tp, 3.000e+16)
self.assertAlmostEqual(mt.clvd, 0.22)
# exporting back to XML should result in the same document
original = open(filename, "rt").read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)
def setUp(self):
# directory where the test files are located
self.path = os.path.join(os.path.dirname(__file__), 'data')
self.neries_filename = os.path.join(self.path, 'neries_events.xml')
self.neries_catalog = readQuakeML(self.neries_filename)
catalog_file = '/Users/escuser/project/catalog/all_6.quakeml'
events_file = '/Users/escuser/project/catalog/all_6.txt'
# This folder must exist:
pick_file_path = '/Users/escuser/project/catalog/all_6'
#Which channels?
channels = ["HNZ", "HLZ", "HHZ", "BHZ"]
# Which stations do you want to get events on?
stations = ['FRD', 'RDM']
################################################################################
#load the catalog
catalog = quakeml.readQuakeML(catalog_file)
print 'Read catalog'
# If you're making pick files...
if make_pick_files:
# Read in the events from the human readable text file; define the pick client
events = genfromtxt(events_file, usecols=5, dtype='S')
pick_client = Client(data_center)
# For every event in the catalog, loop through to find the event id...
for kevent in range(hot_start, len(events)):
id_event = events[kevent]
#if 'ci' in id_event: #SoCla event, get picks, otehrwise ignore
def test_write_with_extra_tags_and_read(self):
"""
Tests that a QuakeML file with additional custom "extra" tags gets
written correctly and that when reading it again the extra tags are
parsed correctly.
"""
filename = os.path.join(self.path, "quakeml_1.2_origin.xml")
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
cat = readQuakeML(filename)
self.assertEqual(len(w), 0)
# add some custom tags to first event:
# - tag with explicit namespace but no explicit ns abbreviation
# - tag without explicit namespace (gets obspy default ns)
# - tag with explicit namespace and namespace abbreviation
my_extra = AttribDict({
'public': {
'value': False,
'namespace': r"http://some-page.de/xmlns/1.0",
'attrib': {
u"some_attrib": u"some_value",
u"another_attrib": u"another_value"
}
},
'custom': {
'value': u"True",
'namespace': r'http://test.org/xmlns/0.1'
},
'new_tag': {
'value': 1234,
'namespace': r"http://test.org/xmlns/0.1"
},
'tX': {
'value': UTCDateTime('2013-01-02T13:12:14.600000Z'),
'namespace': r'http://test.org/xmlns/0.1'
},
'dataid': {
'namespace': r'http://anss.org/xmlns/catalog/0.1',
'type': 'attribute',
'value': '00999999'
}
})
nsmap = {
"ns0": r"http://test.org/xmlns/0.1",
"catalog": r'http://anss.org/xmlns/catalog/0.1'
}
cat[0].extra = my_extra.copy()
# insert a pick with an extra field
p = Pick()
p.extra = {
'weight': {
'value': 2,
'namespace': r"http://test.org/xmlns/0.1"
}
}
cat[0].picks.append(p)
with NamedTemporaryFile() as tf:
tmpfile = tf.name
# write file
cat.write(tmpfile, format="QUAKEML", nsmap=nsmap)
# check contents
with open(tmpfile, "rb") as fh:
# enforce reproducible attribute orders through write_c14n
obj = etree.fromstring(fh.read()).getroottree()
buf = io.BytesIO()
obj.write_c14n(buf)
buf.seek(0, 0)
content = buf.read()
# check namespace definitions in root element
expected = [
b'<q:quakeml',
b'xmlns:catalog="http://anss.org/xmlns/catalog/0.1"',
b'xmlns:ns0="http://test.org/xmlns/0.1"',
b'xmlns:ns1="http://some-page.de/xmlns/1.0"',
b'xmlns:q="http://quakeml.org/xmlns/quakeml/1.2"',
b'xmlns="http://quakeml.org/xmlns/bed/1.2"'
]
for line in expected:
self.assertTrue(line in content)
# check additional tags
expected = [
b'<ns0:custom>True</ns0:custom>',
b'<ns0:new_tag>1234</ns0:new_tag>',
b'<ns0:tX>2013-01-02T13:12:14.600000Z</ns0:tX>',
b'<ns1:public '
b'another_attrib="another_value" '
b'some_attrib="some_value">false</ns1:public>'
]
for line in expected:
self.assertTrue(line in content)
# now, read again to test if it's parsed correctly..
cat = readQuakeML(tmpfile)
# when reading..
# - namespace abbreviations should be disregarded
# - we always end up with a namespace definition, even if it was
# omitted when originally setting the custom tag
# - custom namespace abbreviations should attached to Catalog
self.assertTrue(hasattr(cat[0], "extra"))
def _tostr(x):
if isinstance(x, bool):
if x:
return str("true")
else:
return str("false")
return str(x)
for key, value in my_extra.items():
my_extra[key]['value'] = _tostr(value['value'])
self.assertEqual(cat[0].extra, my_extra)
self.assertTrue(hasattr(cat[0].picks[0], "extra"))
self.assertEqual(cat[0].picks[0].extra, {
'weight': {
'value': '2',
'namespace': r'http://test.org/xmlns/0.1'
}
})
self.assertTrue(hasattr(cat, "nsmap"))
self.assertTrue(getattr(cat, "nsmap")['ns0'] == nsmap['ns0'])
def setUp(self):
self.path = os.path.join(os.path.dirname(__file__), 'data')
qml_file = os.path.join(self.path, 'qml-example-1.2-RC3.xml')
self.c = readQuakeML(qml_file)
self.event = self.c.events[0]
def test_origin(self):
"""
Tests Origin object.
"""
filename = os.path.join(self.path, 'quakeml_1.2_origin.xml')
catalog = readQuakeML(filename)
self.assertEqual(len(catalog), 1)
self.assertEqual(len(catalog[0].origins), 1)
origin = catalog[0].origins[0]
self.assertEqual(
origin.resource_id,
ResourceIdentifier(
'smi:www.iris.edu/ws/event/query?originId=7680412'))
self.assertEqual(origin.time, UTCDateTime("2011-03-11T05:46:24.1200"))
self.assertEqual(origin.latitude, 38.297)
self.assertEqual(origin.latitude_errors.lower_uncertainty, None)
self.assertEqual(origin.longitude, 142.373)
self.assertEqual(origin.longitude_errors.uncertainty, None)
self.assertEqual(origin.depth, 29.0)
self.assertEqual(origin.depth_errors.confidence_level, 50.0)
self.assertEqual(origin.depth_type, "from location")
self.assertEqual(origin.method_id,
ResourceIdentifier(resource_id="smi:some/method/NA"))
self.assertEqual(origin.time_fixed, None)
self.assertEqual(origin.epicenter_fixed, False)
self.assertEqual(
origin.reference_system_id,
ResourceIdentifier(resource_id="smi:some/reference/muh"))
self.assertEqual(origin.earth_model_id,
ResourceIdentifier(resource_id="smi:same/model/maeh"))
self.assertEqual(origin.evaluation_mode, "manual")
self.assertEqual(origin.evaluation_status, "preliminary")
self.assertEqual(origin.origin_type, "hypocenter")
# composite times
self.assertEqual(len(origin.composite_times), 2)
c = origin.composite_times
self.assertEqual(c[0].year, 2029)
self.assertEqual(c[0].month, None)
self.assertEqual(c[0].day, None)
self.assertEqual(c[0].hour, 12)
self.assertEqual(c[0].minute, None)
self.assertEqual(c[0].second, None)
self.assertEqual(c[1].year, None)
self.assertEqual(c[1].month, None)
self.assertEqual(c[1].day, None)
self.assertEqual(c[1].hour, 1)
self.assertEqual(c[1].minute, None)
self.assertEqual(c[1].second, 29.124234)
# quality
self.assertEqual(origin.quality.used_station_count, 16)
self.assertEqual(origin.quality.standard_error, 0)
self.assertEqual(origin.quality.azimuthal_gap, 231)
self.assertEqual(origin.quality.maximum_distance, 53.03)
self.assertEqual(origin.quality.minimum_distance, 2.45)
self.assertEqual(origin.quality.associated_phase_count, None)
self.assertEqual(origin.quality.associated_station_count, None)
self.assertEqual(origin.quality.depth_phase_count, None)
self.assertEqual(origin.quality.secondary_azimuthal_gap, None)
self.assertEqual(origin.quality.ground_truth_level, None)
self.assertEqual(origin.quality.median_distance, None)
# comments
self.assertEqual(len(origin.comments), 2)
c = origin.comments
self.assertEqual(c[0].text, 'Some comment')
self.assertEqual(
c[0].resource_id,
ResourceIdentifier(resource_id="smi:some/comment/reference"))
self.assertEqual(c[0].creation_info.author, 'EMSC')
self.assertEqual(c[1].resource_id, None)
self.assertEqual(c[1].creation_info, None)
self.assertEqual(c[1].text, 'Another comment')
# creation info
self.assertEqual(origin.creation_info.author, "NEIC")
self.assertEqual(origin.creation_info.agency_id, None)
self.assertEqual(origin.creation_info.author_uri, None)
self.assertEqual(origin.creation_info.agency_uri, None)
self.assertEqual(origin.creation_info.creation_time, None)
self.assertEqual(origin.creation_info.version, None)
# origin uncertainty
u = origin.origin_uncertainty
self.assertEqual(u.preferred_description, "uncertainty ellipse")
self.assertEqual(u.horizontal_uncertainty, 9000)
self.assertEqual(u.min_horizontal_uncertainty, 6000)
self.assertEqual(u.max_horizontal_uncertainty, 10000)
self.assertEqual(u.azimuth_max_horizontal_uncertainty, 80.0)
# confidence ellipsoid
c = u.confidence_ellipsoid
self.assertEqual(c.semi_intermediate_axis_length, 2.123)
self.assertEqual(c.major_axis_rotation, 5.123)
self.assertEqual(c.major_axis_plunge, 3.123)
self.assertEqual(c.semi_minor_axis_length, 1.123)
self.assertEqual(c.semi_major_axis_length, 0.123)
self.assertEqual(c.major_axis_azimuth, 4.123)
# exporting back to XML should result in the same document
original = open(filename, "rt").read()
processed = Pickler().dumps(catalog)
self._compareStrings(original, processed)