def test_response_calculation_from_seed_and_xseed(self):
"""
Test the response calculations with the obspy.core interface.
It does it by converting whatever it gets to RESP files and then
uses evalresp to get the response. This is compared to using the
ObsPy Response object - this also uses evalresp but the actual flow
of the data is very different.
This is an expensive test but worth it for the trust it builds and
the bugs it found and prevents.
"""
# Very broad range but the responses should be exactly identical as
# they use the same code under the hood so it should prove no issue.
frequencies = np.logspace(-3, 3, 20)
for filename in self.seed_files + self.xseed_files:
# Parse the files using the Parser object.
with CatchAndAssertWarnings():
p = Parser(filename)
p_resp = {_i[0]: _i[1] for _i in p.get_resp()}
# Also read using the core routines.
inv = obspy.read_inventory(filename)
# Get all the channels and epochs.
channels = collections.defaultdict(list)
for c in p.get_inventory()["channels"]:
channels[c["channel_id"]].append(
(c["start_date"], c["end_date"]))
# Loop over each.
for channel, epochs in channels.items():
with NamedTemporaryFile() as tf:
r = p_resp["RESP.%s" % channel]
r.seek(0, 0)
tf.write(r.read())
# Now loop over the epochs.
for start, end in epochs:
if end:
t = start + (end - start) / 2
else:
t = start + 10
# Find response
n, s, l, c = channel.split(".")
_inv_t = inv.select(network=n, station=s,
location=l, channel=c,
starttime=t - 1, endtime=t + 1)
# Should now only be a single channel.
self.assertEqual(_inv_t.get_contents()["channels"],
[channel])
inv_r = _inv_t[0][0][0].response
for unit in ("DISP", "VEL", "ACC"):
# Directly call evalresp.
e_r = evalresp_for_frequencies(
t_samp=None, frequencies=frequencies,
filename=tf.name, date=t, units=unit)
i_r = inv_r.get_evalresp_response_for_frequencies(
frequencies=frequencies, output=unit)
# Adaptive absolute tolerance to deal with very
# small values.
atol = 1E-7 * max(np.abs(e_r).max(),
np.abs(i_r).max())
np.testing.assert_allclose(
e_r.real, i_r.real,
err_msg="real - %s - %s" % (filename, unit),
rtol=1E-6, atol=atol)
np.testing.assert_allclose(
e_r.imag, i_r.imag,
err_msg="imag - %s - %s" % (filename, unit),
rtol=1E-6, atol=atol)
# Bonus: Also read the RESP file directly with
# obspy.core and test the response.
i_r_r = obspy.read_inventory(tf.name).select(
starttime=t - 1,
endtime=t + 1)[0][0][0].response\
.get_evalresp_response_for_frequencies(
frequencies=frequencies, output=unit)
np.testing.assert_allclose(
e_r.real, i_r_r.real,
err_msg="RESP real - %s - %s" % (filename,
unit),
rtol=1E-6, atol=atol)
np.testing.assert_allclose(
e_r.imag, i_r_r.imag,
err_msg="RESP imag - %s - %s" % (filename,
unit),
rtol=1E-6, atol=atol)
def test_read_resp_metadata(self):
"""
Manually assert that the meta-data is read correctly for all the
RESP files.
"""
# File A
filename_a = os.path.join(self.data_path,
"RESP.BW.FURT..EHZ")
inv = obspy.read_inventory(filename_a)
self.assertEqual(len(inv), 1)
self.assertEqual(len(inv[0]), 1)
self.assertEqual(len(inv[0][0]), 1)
network = inv[0]
station = inv[0][0]
channel = inv[0][0][0]
resp = inv[0][0][0].response
self.assertEqual(network.code, "BW")
self.assertEqual(station.code, "FURT")
self.assertEqual(station.start_date, None)
self.assertEqual(station.end_date, None)
self.assertEqual(channel.location_code, "")
self.assertEqual(channel.code, "EHZ")
self.assertEqual(channel.start_date, obspy.UTCDateTime(2001, 1, 1))
self.assertEqual(channel.end_date, None)
# Also check the input and output units.
self.assertEqual(resp.instrument_sensitivity.input_units, "M/S")
self.assertEqual(resp.instrument_sensitivity.input_units_description,
"Velocity in Meters per Second")
self.assertEqual(resp.instrument_sensitivity.output_units, "COUNTS")
self.assertEqual(resp.instrument_sensitivity.output_units_description,
"Digital Counts")
# File B
filename_b = os.path.join(self.data_path,
"RESP.XX.NR008..HHZ.130.1.100")
inv = obspy.read_inventory(filename_b)
self.assertEqual(len(inv), 1)
self.assertEqual(len(inv[0]), 1)
self.assertEqual(len(inv[0][0]), 1)
network = inv[0]
station = inv[0][0]
channel = inv[0][0][0]
resp = inv[0][0][0].response
self.assertEqual(network.code, "XX")
self.assertEqual(station.code, "NR008")
self.assertEqual(station.start_date, None)
self.assertEqual(station.end_date, None)
self.assertEqual(channel.location_code, "")
self.assertEqual(channel.code, "HHZ")
self.assertEqual(channel.start_date, obspy.UTCDateTime(2006, 1, 1))
self.assertEqual(channel.end_date,
obspy.UTCDateTime(3000, 1, 1) - 1)
# Also check the input and output units.
self.assertEqual(resp.instrument_sensitivity.input_units, "M/S")
self.assertEqual(resp.instrument_sensitivity.input_units_description,
"Velocity in Meters per Second")
self.assertEqual(resp.instrument_sensitivity.output_units, "COUNTS")
self.assertEqual(resp.instrument_sensitivity.output_units_description,
"Digital Counts")
# File C
filename_c = os.path.join(self.data_path,
"RESP.XX.NS085..BHZ.STS2_gen3.120.1500")
inv = obspy.read_inventory(filename_c)
self.assertEqual(len(inv), 1)
self.assertEqual(len(inv[0]), 1)
self.assertEqual(len(inv[0][0]), 1)
network = inv[0]
station = inv[0][0]
channel = inv[0][0][0]
resp = inv[0][0][0].response
self.assertEqual(network.code, "XX")
self.assertEqual(station.code, "NS085")
self.assertEqual(station.start_date, None)
self.assertEqual(station.end_date, None)
self.assertEqual(channel.location_code, "")
self.assertEqual(channel.code, "BHZ")
self.assertEqual(channel.start_date, obspy.UTCDateTime(2006, 1, 1))
self.assertEqual(channel.end_date, None)
# Also check the input and output units.
self.assertEqual(resp.instrument_sensitivity.input_units, "M/S")
self.assertEqual(resp.instrument_sensitivity.input_units_description,
"Velocity in Meters per Second")
self.assertEqual(resp.instrument_sensitivity.output_units, "COUNTS")
self.assertEqual(resp.instrument_sensitivity.output_units_description,
"Digital Counts")
# Merge A + B to get a multi-response file.
with NamedTemporaryFile() as tf:
with io.open(filename_a, "rb") as fh:
tf.write(fh.read())
tf.write(b"\n")
with io.open(filename_b, "rb") as fh:
tf.write(fh.read())
tf.seek(0, 0)
inv = obspy.read_inventory(tf.name)
self.assertEqual(len(inv), 2)
self.assertEqual(len(inv[0]), 1)
self.assertEqual(len(inv[0][0]), 1)
self.assertEqual(len(inv[1]), 1)
self.assertEqual(len(inv[1][0]), 1)
network = inv[0]
station = inv[0][0]
channel = inv[0][0][0]
resp = inv[0][0][0].response
self.assertEqual(network.code, "BW")
self.assertEqual(station.code, "FURT")
self.assertEqual(station.start_date, None)
self.assertEqual(station.end_date, None)
self.assertEqual(channel.location_code, "")
self.assertEqual(channel.code, "EHZ")
self.assertEqual(channel.start_date, obspy.UTCDateTime(2001, 1, 1))
self.assertEqual(channel.end_date, None)
# Also check the input and output units.
self.assertEqual(resp.instrument_sensitivity.input_units, "M/S")
self.assertEqual(resp.instrument_sensitivity.input_units_description,
"Velocity in Meters per Second")
self.assertEqual(resp.instrument_sensitivity.output_units, "COUNTS")
self.assertEqual(resp.instrument_sensitivity.output_units_description,
"Digital Counts")
network = inv[1]
station = inv[1][0]
channel = inv[1][0][0]
resp = inv[1][0][0].response
self.assertEqual(network.code, "XX")
self.assertEqual(station.code, "NR008")
self.assertEqual(station.start_date, None)
self.assertEqual(station.end_date, None)
self.assertEqual(channel.location_code, "")
self.assertEqual(channel.code, "HHZ")
self.assertEqual(channel.start_date, obspy.UTCDateTime(2006, 1, 1))
self.assertEqual(channel.end_date,
obspy.UTCDateTime(3000, 1, 1) - 1)
# Also check the input and output units.
self.assertEqual(resp.instrument_sensitivity.input_units, "M/S")
self.assertEqual(resp.instrument_sensitivity.input_units_description,
"Velocity in Meters per Second")
self.assertEqual(resp.instrument_sensitivity.output_units, "COUNTS")
self.assertEqual(resp.instrument_sensitivity.output_units_description,
"Digital Counts")