def test_read_write_open_file(self):
st = _read_sac(self.file)
with NamedTemporaryFile() as tf_out:
_write_sac(st, tf_out)
tf_out.seek(0, 0)
st2 = _read_sac(tf_out)
tr = st[0]
tr2 = st2[0]
# depmen is different as it is actually calculated on the fly.
del tr.stats.sac.depmen
del tr2.stats.sac.depmen
np.testing.assert_array_equal(tr.data, tr2.data)
self.assertEqual(tr, tr2)
def test_read_write_open_file(self):
st = _read_sac(self.file)
with NamedTemporaryFile() as tf_out:
_write_sac(st, tf_out)
tf_out.seek(0, 0)
st2 = _read_sac(tf_out)
tr = st[0]
tr2 = st2[0]
# depmen is different as it is actually calculated on the fly.
del tr.stats.sac.depmen
del tr2.stats.sac.depmen
np.testing.assert_array_equal(tr.data, tr2.data)
self.assertEqual(tr, tr2)
def test_read_write_bytes_io(self):
"""
Tests reading and writing to and from BytesIO.
"""
st = _read_sac(self.file)
with io.BytesIO() as buf:
_write_sac(st, buf)
buf.seek(0, 0)
# Attempt to read from it.
st2 = _read_sac(buf)
tr = st[0]
tr2 = st2[0]
# depmen is different as it is actually calculated on the fly.
del tr.stats.sac.depmen
del tr2.stats.sac.depmen
np.testing.assert_array_equal(tr.data, tr2.data)
self.assertEqual(tr, tr2)
def test_read_write_bytes_io(self):
"""
Tests reading and writing to and from BytesIO.
"""
st = _read_sac(self.file)
with io.BytesIO() as buf:
_write_sac(st, buf)
buf.seek(0, 0)
# Attempt to read from it.
st2 = _read_sac(buf)
tr = st[0]
tr2 = st2[0]
# depmen is different as it is actually calculated on the fly.
del tr.stats.sac.depmen
del tr2.stats.sac.depmen
np.testing.assert_array_equal(tr.data, tr2.data)
self.assertEqual(tr, tr2)
