Пример #1
0
 def test_writeSLISTFileMultipleTraces(self):
     """
     Write SLIST file test via obspy.core.ascii.writeTSPAIR.
     """
     testfile = os.path.join(self.path, 'data', 'slist_2_traces.ascii')
     stream_orig = readSLIST(testfile)
     with NamedTemporaryFile() as tf:
         tmpfile = tf.name
         # write
         writeSLIST(stream_orig, tmpfile)
         # look at the raw data
         with open(tmpfile, 'rt') as f:
             lines = f.readlines()
         self.assertTrue(lines[0].startswith('TIMESERIES'))
         self.assertTrue('SLIST' in lines[0])
         self.assertEqual(lines[1].strip(), '185\t181\t185\t189\t194\t205')
         # read again
         stream = readSLIST(tmpfile)
     stream.verify()
     # sort traces to ensure comparable results
     stream.sort()
     self.assertEqual(stream[0].stats.network, 'XX')
     self.assertEqual(stream[0].stats.station, 'TEST')
     self.assertEqual(stream[0].stats.location, '')
     self.assertEqual(stream[0].stats.channel, 'BHE')
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 630)
     self.assertEqual(stream[0].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     self.assertEqual(stream[0].stats.mseed.dataquality, 'R')
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[0].data[0:4], data)
     # check last 4 samples
     data = [781, 785, 778, 772]
     np.testing.assert_array_almost_equal(stream[0].data[-4:], data)
     # second trace
     self.assertEqual(stream[1].stats.network, 'XX')
     self.assertEqual(stream[1].stats.station, 'TEST')
     self.assertEqual(stream[1].stats.location, '')
     self.assertEqual(stream[1].stats.channel, 'BHZ')
     self.assertEqual(stream[1].stats.sampling_rate, 40.0)
     self.assertEqual(stream[1].stats.npts, 635)
     self.assertEqual(stream[1].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[1].stats.calib, 1.0e-00)
     self.assertEqual(stream[0].stats.mseed.dataquality, 'R')
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[1].data[0:4], data)
     # check last 4 samples
     data = [761, 755, 748, 746]
     np.testing.assert_array_almost_equal(stream[1].data[-4:], data)
Пример #2
0
 def test_writeSLISTFileMultipleTraces(self):
     """
     Write SLIST file test via obspy.core.ascii.writeTSPAIR.
     """
     testfile = os.path.join(self.path, 'data', 'slist_2_traces.ascii')
     stream_orig = readSLIST(testfile)
     tmpfile = NamedTemporaryFile().name
     # write
     writeSLIST(stream_orig, tmpfile)
     # look at the raw data
     lines = open(tmpfile, 'rt').readlines()
     self.assertTrue(lines[0].startswith('TIMESERIES'))
     self.assertTrue('SLIST' in lines[0])
     self.assertEqual(lines[1].strip(), '185\t181\t185\t189\t194\t205')
     # read again
     stream = readSLIST(tmpfile)
     stream.verify()
     # sort traces to ensure comparable results
     stream.sort()
     self.assertEqual(stream[0].stats.network, 'XX')
     self.assertEqual(stream[0].stats.station, 'TEST')
     self.assertEqual(stream[0].stats.location, '')
     self.assertEqual(stream[0].stats.channel, 'BHE')
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 630)
     self.assertEqual(stream[0].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     self.assertEqual(stream[0].stats.mseed.dataquality, 'R')
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[0].data[0:4], data)
     # check last 4 samples
     data = [781, 785, 778, 772]
     np.testing.assert_array_almost_equal(stream[0].data[-4:], data)
     # second trace
     self.assertEqual(stream[1].stats.network, 'XX')
     self.assertEqual(stream[1].stats.station, 'TEST')
     self.assertEqual(stream[1].stats.location, '')
     self.assertEqual(stream[1].stats.channel, 'BHZ')
     self.assertEqual(stream[1].stats.sampling_rate, 40.0)
     self.assertEqual(stream[1].stats.npts, 635)
     self.assertEqual(stream[1].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[1].stats.calib, 1.0e-00)
     self.assertEqual(stream[0].stats.mseed.dataquality, 'R')
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[1].data[0:4], data)
     # check last 4 samples
     data = [761, 755, 748, 746]
     np.testing.assert_array_almost_equal(stream[1].data[-4:], data)
     # clean up
     os.remove(tmpfile)
Пример #3
0
 def test_writeSLIST(self):
     """
     Write SLIST file test via obspy.core.ascii.writeTSPAIR.
     """
     # float32
     testfile = os.path.join(self.path, 'data', 'slist_float.ascii')
     stream_orig = readSLIST(testfile)
     with NamedTemporaryFile() as tf:
         tmpfile = tf.name
         # write
         writeSLIST(stream_orig, tmpfile)
         # look at the raw data
         with open(tmpfile, 'rt') as f:
             lines = f.readlines()
         self.assertEqual(
             lines[0].strip(),
             'TIMESERIES XX_TEST__BHZ_R, 12 samples, 40 sps, ' +
             '2008-01-15T00:00:00.025000, SLIST, FLOAT, Counts')
         self.assertEqual(
             lines[1].strip(),
             '185.009995\t181.020004\t185.029999\t189.039993\t' +
             '194.050003\t205.059998')
         # read again
         stream = readSLIST(tmpfile)
         stream.verify()
         self.assertEqual(stream[0].stats.network, 'XX')
         self.assertEqual(stream[0].stats.station, 'TEST')
         self.assertEqual(stream[0].stats.location, '')
         self.assertEqual(stream[0].stats.channel, 'BHZ')
         self.assertEqual(stream[0].stats.sampling_rate, 40.0)
         self.assertEqual(stream[0].stats.npts, 12)
         self.assertEqual(stream[0].stats.starttime,
                          UTCDateTime("2008-01-15T00:00:00.025000"))
         self.assertEqual(stream[0].stats.calib, 1.0e-00)
         self.assertEqual(stream[0].stats.mseed.dataquality, 'R')
         data = [
             185.01, 181.02, 185.03, 189.04, 194.05, 205.06, 209.07, 214.08,
             222.09, 225.98, 226.99, 219.00
         ]
         np.testing.assert_array_almost_equal(stream[0].data,
                                              data,
                                              decimal=2)
         # compare raw header
         with open(testfile, 'rt') as f:
             lines_orig = f.readlines()
         with open(tmpfile, 'rt') as f:
             lines_new = f.readlines()
     self.assertEqual(lines_orig[0], lines_new[0])
Пример #4
0
 def test_writeSLIST(self):
     """
     Write SLIST file test via obspy.core.ascii.writeTSPAIR.
     """
     # float32
     testfile = os.path.join(self.path, 'data', 'slist_float.ascii')
     stream_orig = readSLIST(testfile)
     with NamedTemporaryFile() as tf:
         tmpfile = tf.name
         # write
         writeSLIST(stream_orig, tmpfile)
         # look at the raw data
         with open(tmpfile, 'rt') as f:
             lines = f.readlines()
         self.assertEqual(
             lines[0].strip(),
             'TIMESERIES XX_TEST__BHZ_R, 12 samples, 40 sps, ' +
             '2008-01-15T00:00:00.025000, SLIST, FLOAT, Counts')
         self.assertEqual(
             lines[1].strip(),
             '185.009995\t181.020004\t185.029999\t189.039993\t' +
             '194.050003\t205.059998')
         # read again
         stream = readSLIST(tmpfile)
         stream.verify()
         self.assertEqual(stream[0].stats.network, 'XX')
         self.assertEqual(stream[0].stats.station, 'TEST')
         self.assertEqual(stream[0].stats.location, '')
         self.assertEqual(stream[0].stats.channel, 'BHZ')
         self.assertEqual(stream[0].stats.sampling_rate, 40.0)
         self.assertEqual(stream[0].stats.npts, 12)
         self.assertEqual(stream[0].stats.starttime,
                          UTCDateTime("2008-01-15T00:00:00.025000"))
         self.assertEqual(stream[0].stats.calib, 1.0e-00)
         self.assertEqual(stream[0].stats.mseed.dataquality, 'R')
         data = [185.01, 181.02, 185.03, 189.04, 194.05, 205.06,
                 209.07, 214.08, 222.09, 225.98, 226.99, 219.00]
         np.testing.assert_array_almost_equal(stream[0].data, data,
                                              decimal=2)
         # compare raw header
         with open(testfile, 'rt') as f:
             lines_orig = f.readlines()
         with open(tmpfile, 'rt') as f:
             lines_new = f.readlines()
     self.assertEqual(lines_orig[0], lines_new[0])
Пример #5
0
 def test_writeSLIST(self):
     """
     Write SLIST file test via obspy.core.ascii.writeTSPAIR.
     """
     # float32
     testfile = os.path.join(self.path, "data", "slist_float.ascii")
     stream_orig = readSLIST(testfile)
     tmpfile = NamedTemporaryFile().name
     # write
     writeSLIST(stream_orig, tmpfile)
     # look at the raw data
     lines = open(tmpfile, "rt").readlines()
     self.assertEquals(
         lines[0].strip(),
         "TIMESERIES XX_TEST__BHZ_R, 12 samples, 40 sps, " + "2008-01-15T00:00:00.025000, SLIST, FLOAT, Counts",
     )
     self.assertEquals(
         lines[1].strip(), "185.009995\t181.020004\t185.029999\t189.039993\t194.050003\t" + "205.059998"
     )
     # read again
     stream = readSLIST(tmpfile)
     stream.verify()
     self.assertEqual(stream[0].stats.network, "XX")
     self.assertEqual(stream[0].stats.station, "TEST")
     self.assertEqual(stream[0].stats.location, "")
     self.assertEqual(stream[0].stats.channel, "BHZ")
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 12)
     self.assertEqual(stream[0].stats.starttime, UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     self.assertEqual(stream[0].stats.mseed.dataquality, "R")
     data = [185.01, 181.02, 185.03, 189.04, 194.05, 205.06, 209.07, 214.08, 222.09, 225.98, 226.99, 219.00]
     np.testing.assert_array_almost_equal(stream[0].data, data, decimal=2)
     # compare raw header
     lines_orig = open(testfile, "rt").readlines()
     lines_new = open(tmpfile, "rt").readlines()
     self.assertEqual(lines_orig[0], lines_new[0])
     # clean up
     os.remove(tmpfile)
Пример #6
0
 def test_readSLISTFileHeadOnly(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     testfile = os.path.join(self.path, "data", "slist.ascii")
     # read
     stream = readSLIST(testfile, headonly=True)
     self.assertEqual(stream[0].stats.network, "XX")
     self.assertEqual(stream[0].stats.station, "TEST")
     self.assertEqual(stream[0].stats.location, "")
     self.assertEqual(stream[0].stats.channel, "BHZ")
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 635)
     self.assertEqual(stream[0].stats.starttime, UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     self.assertEqual(len(stream[0].data), 0)
Пример #7
0
 def test_readSLISTFileHeadOnly(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     testfile = os.path.join(self.path, 'data', 'slist.ascii')
     # read
     stream = readSLIST(testfile, headonly=True)
     self.assertEqual(stream[0].stats.network, 'XX')
     self.assertEqual(stream[0].stats.station, 'TEST')
     self.assertEqual(stream[0].stats.location, '')
     self.assertEqual(stream[0].stats.channel, 'BHZ')
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 635)
     self.assertEqual(stream[0].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     self.assertEqual(len(stream[0].data), 0)
Пример #8
0
 def test_readSLISTFileEncoding(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     # float32
     testfile = os.path.join(self.path, "data", "slist_float.ascii")
     stream = readSLIST(testfile)
     self.assertEqual(stream[0].stats.network, "XX")
     self.assertEqual(stream[0].stats.station, "TEST")
     self.assertEqual(stream[0].stats.location, "")
     self.assertEqual(stream[0].stats.channel, "BHZ")
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 12)
     self.assertEqual(stream[0].stats.starttime, UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     data = [185.01, 181.02, 185.03, 189.04, 194.05, 205.06, 209.07, 214.08, 222.09, 225.98, 226.99, 219.00]
     np.testing.assert_array_almost_equal(stream[0].data, data, decimal=2)
     # unknown encoding
     testfile = os.path.join(self.path, "data", "slist_unknown.ascii")
     self.assertRaises(NotImplementedError, readSLIST, testfile)
Пример #9
0
 def test_readSLISTFileMultipleTraces(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     testfile = os.path.join(self.path, 'data', 'slist_2_traces.ascii')
     # read
     stream = readSLIST(testfile)
     stream.verify()
     self.assertEqual(stream[0].stats.network, 'XX')
     self.assertEqual(stream[0].stats.station, 'TEST')
     self.assertEqual(stream[0].stats.location, '')
     self.assertEqual(stream[0].stats.channel, 'BHZ')
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 635)
     self.assertEqual(stream[0].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[0].data[0:4], data)
     # check last 4 samples
     data = [761, 755, 748, 746]
     np.testing.assert_array_almost_equal(stream[0].data[-4:], data)
     # second trace
     self.assertEqual(stream[1].stats.network, 'XX')
     self.assertEqual(stream[1].stats.station, 'TEST')
     self.assertEqual(stream[1].stats.location, '')
     self.assertEqual(stream[1].stats.channel, 'BHE')
     self.assertEqual(stream[1].stats.sampling_rate, 40.0)
     self.assertEqual(stream[1].stats.npts, 630)
     self.assertEqual(stream[1].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[1].stats.calib, 1.0e-00)
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[1].data[0:4], data)
     # check last 4 samples
     data = [781, 785, 778, 772]
     np.testing.assert_array_almost_equal(stream[1].data[-4:], data)
Пример #10
0
 def test_readSLISTFileMultipleTraces(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     testfile = os.path.join(self.path, 'data', 'slist_2_traces.ascii')
     # read
     stream = readSLIST(testfile)
     stream.verify()
     self.assertEqual(stream[0].stats.network, 'XX')
     self.assertEqual(stream[0].stats.station, 'TEST')
     self.assertEqual(stream[0].stats.location, '')
     self.assertEqual(stream[0].stats.channel, 'BHZ')
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 635)
     self.assertEqual(stream[0].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[0].data[0:4], data)
     # check last 4 samples
     data = [761, 755, 748, 746]
     np.testing.assert_array_almost_equal(stream[0].data[-4:], data)
     # second trace
     self.assertEqual(stream[1].stats.network, 'XX')
     self.assertEqual(stream[1].stats.station, 'TEST')
     self.assertEqual(stream[1].stats.location, '')
     self.assertEqual(stream[1].stats.channel, 'BHE')
     self.assertEqual(stream[1].stats.sampling_rate, 40.0)
     self.assertEqual(stream[1].stats.npts, 630)
     self.assertEqual(stream[1].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[1].stats.calib, 1.0e-00)
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[1].data[0:4], data)
     # check last 4 samples
     data = [781, 785, 778, 772]
     np.testing.assert_array_almost_equal(stream[1].data[-4:], data)
Пример #11
0
 def test_readSLISTFileEncoding(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     # float32
     testfile = os.path.join(self.path, 'data', 'slist_float.ascii')
     stream = readSLIST(testfile)
     self.assertEqual(stream[0].stats.network, 'XX')
     self.assertEqual(stream[0].stats.station, 'TEST')
     self.assertEqual(stream[0].stats.location, '')
     self.assertEqual(stream[0].stats.channel, 'BHZ')
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 12)
     self.assertEqual(stream[0].stats.starttime,
                      UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     data = [185.01, 181.02, 185.03, 189.04, 194.05, 205.06,
             209.07, 214.08, 222.09, 225.98, 226.99, 219.00]
     np.testing.assert_array_almost_equal(stream[0].data, data, decimal=2)
     # unknown encoding
     testfile = os.path.join(self.path, 'data', 'slist_unknown.ascii')
     self.assertRaises(NotImplementedError, readSLIST, testfile)
Пример #12
0
 def test_readSLISTFileSingleTrace(self):
     """
     Read SLIST file test via obspy.core.ascii.readSLIST.
     """
     testfile = os.path.join(self.path, "data", "slist.ascii")
     # read
     stream = readSLIST(testfile)
     stream.verify()
     self.assertEqual(stream[0].stats.network, "XX")
     self.assertEqual(stream[0].stats.station, "TEST")
     self.assertEqual(stream[0].stats.location, "")
     self.assertEqual(stream[0].stats.channel, "BHZ")
     self.assertEqual(stream[0].stats.sampling_rate, 40.0)
     self.assertEqual(stream[0].stats.npts, 635)
     self.assertEqual(stream[0].stats.starttime, UTCDateTime("2008-01-15T00:00:00.025000"))
     self.assertEqual(stream[0].stats.calib, 1.0e-00)
     # check first 4 samples
     data = [185, 181, 185, 189]
     np.testing.assert_array_almost_equal(stream[0].data[0:4], data)
     # check last 4 samples
     data = [761, 755, 748, 746]
     np.testing.assert_array_almost_equal(stream[0].data[-4:], data)