def test_readWriteXY(self):
"""
Tests for ascii sac io
"""
with NamedTemporaryFile() as tf:
tempfile = tf.name
tfile = os.path.join(os.path.dirname(__file__), 'data', 'test.sac')
with open(tfile, "rb") as fh:
t = SacIO(fh)
with open(tempfile, "wb") as fh:
t.write_sac_xy(fh)
with open(tempfile, "rb") as fh:
d = SacIO(fh, alpha=True)
e = SacIO()
with open(tempfile, "rb") as fh:
e.read_sac_xy(fh)
self.assertEqual(e.get_header_value('npts'),
d.get_header_value('npts'))
with open(tempfile, "rb") as fh:
self.assertEqual(e.is_valid_xy_sac_file(fh), True)
with open(tempfile, "rb") as fh:
self.assertEqual(e.is_valid_sac_file(fh), False)
with NamedTemporaryFile() as tf:
tempfile = tf.name
with open(tempfile, "wb") as fh:
d.write_sac_binary(fh)
size1 = os.stat(tempfile)[6]
size2 = os.stat(tfile)[6]
self.assertEqual(size1, size2)
np.testing.assert_array_almost_equal(t.seis, d.seis, decimal=5)
def test_readWriteXY(self):
"""
Tests for ascii sac io
"""
with NamedTemporaryFile() as tf:
tempfile = tf.name
tfile = os.path.join(os.path.dirname(__file__), 'data', 'test.sac')
with open(tfile, "rb") as fh:
t = SacIO(fh)
with open(tempfile, "wb") as fh:
t.write_sac_xy(fh)
with open(tempfile, "rb") as fh:
d = SacIO(fh, alpha=True)
e = SacIO()
with open(tempfile, "rb") as fh:
e.read_sac_xy(fh)
self.assertEqual(e.get_header_value('npts'),
d.get_header_value('npts'))
with open(tempfile, "rb") as fh:
self.assertEqual(e.is_valid_xy_sac_file(fh), True)
with open(tempfile, "rb") as fh:
self.assertEqual(e.is_valid_sac_file(fh), False)
with NamedTemporaryFile() as tf:
tempfile = tf.name
with open(tempfile, "wb") as fh:
d.write_sac_binary(fh)
size1 = os.stat(tempfile)[6]
size2 = os.stat(tfile)[6]
self.assertEqual(size1, size2)
np.testing.assert_array_almost_equal(t.seis, d.seis, decimal=5)
def test_readXYheader(self):
tfile = os.path.join(os.path.dirname(__file__), 'data', 'test.sac')
with NamedTemporaryFile() as tf:
tempfile = tf.name
with open(tfile, "rb") as fh:
t = SacIO(fh)
with open(tempfile, 'wb') as fh:
t.write_sac_xy(fh)
with open(tempfile, "rb") as fh:
d = SacIO(fh, alpha=True)
e = SacIO()
with open(tempfile, "rb") as fh:
e.read_sac_xy_header(fh)
self.assertEqual(e.get_header_value('npts'),
d.get_header_value('npts'))
self.assertEqual(e.get_header_value('depmen'),
d.get_header_value('depmen'))
self.assertEqual(e.starttime, d.starttime)
self.assertNotEqual(e.seis.size, d.seis.size)
with open(tempfile, "rb") as fh:
c = SacIO(fh, alpha=True, headonly=True)
self.assertEqual(e.seis.size, c.seis.size)
def test_readXYheader(self):
tfile = os.path.join(os.path.dirname(__file__), 'data', 'test.sac')
with NamedTemporaryFile() as tf:
tempfile = tf.name
with open(tfile, "rb") as fh:
t = SacIO(fh)
with open(tempfile, 'wb') as fh:
t.write_sac_xy(fh)
with open(tempfile, "rb") as fh:
d = SacIO(fh, alpha=True)
e = SacIO()
with open(tempfile, "rb") as fh:
e.read_sac_xy_header(fh)
self.assertEqual(e.get_header_value('npts'),
d.get_header_value('npts'))
self.assertEqual(e.get_header_value('depmen'),
d.get_header_value('depmen'))
self.assertEqual(e.starttime, d.starttime)
self.assertNotEqual(e.seis.size, d.seis.size)
with open(tempfile, "rb") as fh:
c = SacIO(fh, alpha=True, headonly=True)
self.assertEqual(e.seis.size, c.seis.size)