def test_readBigEnd(self):
"""
Test reading big endian binary files
"""
tfilel = os.path.join(os.path.dirname(__file__), 'data', 'test.sac')
tfileb = os.path.join(os.path.dirname(__file__), 'data',
'test.sac.swap')
tl = SacIO(tfilel)
tb = SacIO(tfileb)
self.assertEqual(tl.GetHvalue('kevnm'), tb.GetHvalue('kevnm'))
self.assertEqual(tl.GetHvalue('npts'), tb.GetHvalue('npts'))
self.assertEqual(tl.GetHvalueFromFile(tfilel, 'kcmpnm'),
tb.GetHvalueFromFile(tfileb, 'kcmpnm'))
np.testing.assert_array_equal(tl.seis, tb.seis)
def test_readWrite(self):
"""
Tests for SacIO read and write
"""
sacfile = os.path.join(self.path, 'test.sac')
with NamedTemporaryFile() as tf:
tempfile = tf.name
t = SacIO()
t.ReadSacFile(sacfile)
self.assertEqual(t.GetHvalue('npts'), 100)
self.assertEqual(t.GetHvalue("kcmpnm"), "Q ")
self.assertEqual(t.GetHvalue("kstnm"), "STA ")
t.SetHvalue("kstnm", "spiff")
self.assertEqual(t.GetHvalue('kstnm'), 'spiff ')
t.WriteSacBinary(tempfile)
self.assertEqual(os.stat(sacfile)[6], os.stat(tempfile)[6])
self.assertEqual(os.path.exists(tempfile), True)
t.ReadSacHeader(tempfile)
self.assertEqual((t.hf is not None), True)
t.SetHvalue("kstnm", "spoff")
self.assertEqual(t.GetHvalue('kstnm'), 'spoff ')
t.WriteSacHeader(tempfile)
t.SetHvalueInFile(tempfile, "kcmpnm", 'Z ')
self.assertEqual(t.GetHvalueFromFile(tempfile, "kcmpnm"),
'Z ')
self.assertEqual(
SacIO(tempfile, headonly=True).GetHvalue('kcmpnm'), 'Z ')
self.assertEqual(t.IsValidSacFile(tempfile), True)
self.assertEqual(t.IsValidXYSacFile(tempfile), False)
self.assertEqual(SacIO().GetHvalueFromFile(sacfile, 'npts'), 100)
self.assertEqual(SacIO(sacfile).GetHvalue('npts'), 100)
def test_swapbytes(self):
tfilel = os.path.join(os.path.dirname(__file__), 'data', 'test.sac')
tfileb = os.path.join(os.path.dirname(__file__), 'data',
'test.sac.swap')
with NamedTemporaryFile() as tf:
tempfile = tf.name
tb = SacIO(tfileb)
tb.swap_byte_order()
tb.WriteSacBinary(tempfile)
tr1 = SacIO(tempfile)
tl = SacIO(tfilel)
np.testing.assert_array_equal(tl.seis, tr1.seis)
self.assertEqual(tl.GetHvalue('kevnm'), tr1.GetHvalue('kevnm'))
self.assertEqual(tl.GetHvalue('npts'), tr1.GetHvalue('npts'))
self.assertEqual(tl.GetHvalueFromFile(tfilel, 'kcmpnm'),
tr1.GetHvalueFromFile(tempfile, 'kcmpnm'))
