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)