def test_saveWaveform(self):
"""
Testing simple waveform file save method.
"""
# file identical to file retrieved via web interface
client = Client()
start = UTCDateTime("2010-02-27T06:30:00")
end = UTCDateTime("2010-02-27T06:31:00")
origfile = os.path.join(self.path, "data", "IU.ANMO.00.BHZ.mseed")
tempfile = NamedTemporaryFile().name
client.saveWaveform(tempfile, "IU", "ANMO", "00", "BHZ", start, end)
self.assertTrue(filecmp.cmp(origfile, tempfile))
os.remove(tempfile)
# no data raises an exception
self.assertRaises(Exception, client.saveWaveform, "YY", "XXXX", "00", "BHZ", start, end)
def test_saveWaveformToStringIO(self):
"""
Same as test_saveWaveform but saves to a StringIO.
"""
# file identical to file retrieved via web interface
client = Client()
start = UTCDateTime("2010-02-27T06:30:00")
end = UTCDateTime("2010-02-27T06:31:00")
memfile = StringIO.StringIO()
client.saveWaveform(memfile, "IU", "ANMO", "00", "BHZ", start, end)
memfile.seek(0, 0)
new_data = memfile.read()
origfile = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.mseed')
with open(origfile, "rb") as fh:
org_data = fh.read()
self.assertEqual(new_data, org_data)
def test_saveWaveform(self):
"""
Testing simple waveform file save method.
"""
# file identical to file retrieved via web interface
client = Client()
start = UTCDateTime("2010-02-27T06:30:00")
end = UTCDateTime("2010-02-27T06:31:00")
origfile = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.mseed')
tempfile = NamedTemporaryFile().name
client.saveWaveform(tempfile, "IU", "ANMO", "00", "BHZ", start, end)
self.assertTrue(filecmp.cmp(origfile, tempfile))
os.remove(tempfile)
# no data raises an exception
self.assertRaises(Exception, client.saveWaveform, "YY", "XXXX", "00",
"BHZ", start, end)
def test_saveWaveformToStringIO(self):
"""
Same as test_saveWaveform but saves to a StringIO.
"""
# file identical to file retrieved via web interface
client = Client()
start = UTCDateTime("2010-02-27T06:30:00")
end = UTCDateTime("2010-02-27T06:31:00")
memfile = StringIO.StringIO()
client.saveWaveform(memfile, "IU", "ANMO", "00", "BHZ", start, end)
memfile.seek(0, 0)
new_data = memfile.read()
origfile = os.path.join(self.path, 'data', 'IU.ANMO.00.BHZ.mseed')
with open(origfile, "rb") as fh:
org_data = fh.read()
self.assertEqual(new_data, org_data)
