def test_saveWaveform(self):
"""
Tests saveWaveform method.
"""
# initialize client
client = Client("pele.ess.washington.edu", 16017)
start = UTCDateTime(2013, 1, 17)
end = start + 30
with NamedTemporaryFile() as tf:
testfile = tf.name
# 1 channel, cleanup (using SLIST to avoid dependencies)
client.saveWaveform(testfile, 'UW', 'TUCA', '', 'BHZ', start, end,
format="SLIST")
stream = read(testfile)
self.assertEqual(len(stream), 1)
delta = stream[0].stats.delta
trace = stream[0]
self.assertTrue(len(trace) == 1201)
self.assertTrue(trace.stats.starttime >= start - delta)
self.assertTrue(trace.stats.starttime <= start + delta)
self.assertTrue(trace.stats.endtime >= end - delta)
self.assertTrue(trace.stats.endtime <= end + delta)
self.assertEqual(trace.stats.network, 'UW')
self.assertEqual(trace.stats.station, 'TUCA')
self.assertEqual(trace.stats.location, '')
self.assertEqual(trace.stats.channel, 'BHZ')
def test_getWaveform(self):
"""
Tests getWaveform method.
"""
client = Client("pele.ess.washington.edu", 16017)
start = UTCDateTime(2013, 1, 17)
end = start + 30
# example 1 -- 1 channel, cleanup
stream = client.getWaveform('UW', 'TUCA', '', 'BHZ', start, end)
self.assertEqual(len(stream), 1)
delta = stream[0].stats.delta
trace = stream[0]
self.assertTrue(len(trace) == 1201)
self.assertTrue(trace.stats.starttime >= start - delta)
self.assertTrue(trace.stats.starttime <= start + delta)
self.assertTrue(trace.stats.endtime >= end - delta)
self.assertTrue(trace.stats.endtime <= end + delta)
self.assertEqual(trace.stats.network, 'UW')
self.assertEqual(trace.stats.station, 'TUCA')
self.assertEqual(trace.stats.location, '')
self.assertEqual(trace.stats.channel, 'BHZ')
# example 2 -- 1 channel, no cleanup
stream = client.getWaveform('UW', 'TUCA', '', 'BHZ', start, end,
cleanup=False)
self.assertTrue(len(stream) >= 2)
summed_length = array([len(tr) for tr in stream]).sum()
self.assertTrue(summed_length == 1201)
self.assertTrue(stream[0].stats.starttime >= start - delta)
self.assertTrue(stream[0].stats.starttime <= start + delta)
self.assertTrue(stream[-1].stats.endtime >= end - delta)
self.assertTrue(stream[-1].stats.endtime <= end + delta)
for trace in stream:
self.assertEqual(trace.stats.network, 'UW')
self.assertEqual(trace.stats.station, 'TUCA')
self.assertEqual(trace.stats.location, '')
self.assertEqual(trace.stats.channel, 'BHZ')
# example 3 -- component wildcarded with '?'
stream = client.getWaveform('UW', 'TUCA', '', 'BH?', start, end)
self.assertEqual(len(stream), 3)
for trace in stream:
self.assertTrue(len(trace) == 1201)
self.assertTrue(trace.stats.starttime >= start - delta)
self.assertTrue(trace.stats.starttime <= start + delta)
self.assertTrue(trace.stats.endtime >= end - delta)
self.assertTrue(trace.stats.endtime <= end + delta)
self.assertEqual(trace.stats.network, 'UW')
self.assertEqual(trace.stats.station, 'TUCA')
self.assertEqual(trace.stats.location, '')
self.assertEqual(stream[0].stats.channel, 'BHZ')
self.assertEqual(stream[1].stats.channel, 'BHN')
self.assertEqual(stream[2].stats.channel, 'BHE')
def __init__(self,
name='earthworm waveserver client',
host='localhost',
port=16022,
**kwargs):
WaveClient.__init__(self, name=name, **kwargs)
# The Earthworm waveserver host to which the client should connect.
self.host = host
# The port on which the Eartworm waveserver is running on host.
self.port = port
# The obspy earthworm waveserver client instance.
self.client = Client(self.host, self.port, timeout=2)
def setUp(self):
# Monkey patch: set lower default precision of all UTCDateTime objects
UTCDateTime.DEFAULT_PRECISION = 4
self.client = Client("pele.ess.washington.edu", 16017, timeout=7)
def test_availability(self):
client = Client("pele.ess.washington.edu", 16017)
data = client.availability()
seeds = ["%s.%s.%s.%s" % (d[0], d[1], d[2], d[3]) for d in data]
self.assertTrue('UW.TUCA.--.BHZ' in seeds)
