def test_read_sac_from_open_file(self):
"""
Tests reading from an open file.
"""
with open(self.file, "rb") as fh:
# Attempt to read from it.
tr = _read_sac(fh)[0]
# Open file normally and make sure the results are identical.
tr2 = _read_sac(self.file)[0]
np.testing.assert_array_equal(tr.data, tr2.data)
self.assertEqual(tr, tr2)
def test_read_sac_from_open_file(self):
"""
Tests reading from an open file.
"""
with open(self.file, "rb") as fh:
# Attempt to read from it.
tr = _read_sac(fh)[0]
# Open file normally and make sure the results are identical.
tr2 = _read_sac(self.file)[0]
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_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_sac_from_bytes_io(self):
"""
Tests reading from a BytesIO object.
"""
with io.BytesIO() as buf:
# Read file to BytesIO.
with open(self.file, "rb") as fh:
buf.write(fh.read())
buf.seek(0, 0)
# Attempt to read from it.
tr = _read_sac(buf)[0]
# Open file normally and make sure the results are identical.
tr2 = _read_sac(self.file)[0]
np.testing.assert_array_equal(tr.data, tr2.data)
self.assertEqual(tr, tr2)
def test_read_sac_from_bytes_io(self):
"""
Tests reading from a BytesIO object.
"""
with io.BytesIO() as buf:
# Read file to BytesIO.
with open(self.file, "rb") as fh:
buf.write(fh.read())
buf.seek(0, 0)
# Attempt to read from it.
tr = _read_sac(buf)[0]
# Open file normally and make sure the results are identical.
tr2 = _read_sac(self.file)[0]
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)