def test_parse_usgs_param_file_from_bytes_io_and_open_files():
"""
Tests parsing a USGS file from a BytesIO stream and open files..
"""
with io.open(USGS_PARAM_FILE1, "rb") as fh:
finitesource = FiniteSource.from_usgs_param_file(fh)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
7.9374427577095901)
assert finitesource.npointsources == 121
with io.open(USGS_PARAM_FILE1, "rb") as fh:
with io.BytesIO(fh.read()) as buf:
finitesource = FiniteSource.from_usgs_param_file(buf)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
7.9374427577095901)
assert finitesource.npointsources == 121
with io.open(USGS_PARAM_FILE2, "rb") as fh:
finitesource = FiniteSource.from_usgs_param_file(fh)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.26413197488)
assert finitesource.npointsources == 400
with io.open(USGS_PARAM_FILE2, "rb") as fh:
with io.BytesIO(fh.read()) as buf:
finitesource = FiniteSource.from_usgs_param_file(buf)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.26413197488)
assert finitesource.npointsources == 400
def test_parse_usgs_param_file_from_bytes_io_and_open_files():
"""
Tests parsing a USGS file from a BytesIO stream and open files..
"""
with io.open(USGS_PARAM_FILE1, "rb") as fh:
finitesource = FiniteSource.from_usgs_param_file(fh)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
7.8707760910429236)
assert finitesource.npointsources == 121
with io.open(USGS_PARAM_FILE1, "rb") as fh:
with io.BytesIO(fh.read()) as buf:
finitesource = FiniteSource.from_usgs_param_file(buf)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
7.87077609)
assert finitesource.npointsources == 121
with io.open(USGS_PARAM_FILE2, "rb") as fh:
finitesource = FiniteSource.from_usgs_param_file(fh)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.1974653082088)
assert finitesource.npointsources == 400
with io.open(USGS_PARAM_FILE2, "rb") as fh:
with io.BytesIO(fh.read()) as buf:
finitesource = FiniteSource.from_usgs_param_file(buf)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.19746530820887)
assert finitesource.npointsources == 400
def test_parsing_empty_usgs_file():
"""
Parsing an empty USGS file should fail.
"""
with pytest.raises(USGSParamFileParsingException) as e:
FiniteSource.from_usgs_param_file(USGS_PARAM_FILE_EMPTY)
assert e.value.args[0] == 'No point sources found in the file.'
def test_parsing_empty_usgs_file():
"""
Parsing an empty USGS file should fail.
"""
with pytest.raises(USGSParamFileParsingException) as e:
FiniteSource.from_usgs_param_file(USGS_PARAM_FILE_EMPTY)
assert e.value.args[0] == 'No point sources found in the file.'
def test_parse_usgs_param_file():
"""
Tests parsing from a .param file.
"""
# single segment file
finitesource = FiniteSource.from_usgs_param_file(USGS_PARAM_FILE1)
np.testing.assert_almost_equal(finitesource.moment_magnitude, 7.87077609)
assert finitesource.npointsources == 121
# multi segment file
finitesource = FiniteSource.from_usgs_param_file(USGS_PARAM_FILE2)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.1974653082088)
assert finitesource.npointsources == 400
def test_parse_usgs_param_file():
"""
Tests parsing from a .param file.
"""
# single segment file
finitesource = FiniteSource.from_usgs_param_file(USGS_PARAM_FILE1)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
7.87077609)
assert finitesource.npointsources == 121
# multi segment file
finitesource = FiniteSource.from_usgs_param_file(USGS_PARAM_FILE2)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.1974653082088)
assert finitesource.npointsources == 400
def test_parse_usgs_param_file():
"""
Tests parsing from a .param file.
"""
# single segment file
finitesource = FiniteSource.from_usgs_param_file(USGS_PARAM_FILE1)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
7.9374427577095901)
assert finitesource.npointsources == 121
# multi segment file
finitesource = FiniteSource.from_usgs_param_file(USGS_PARAM_FILE2)
np.testing.assert_almost_equal(finitesource.moment_magnitude,
8.26413197488)
assert finitesource.npointsources == 400
def on_open_srf_file_button_released(self):
pwd = os.getcwd()
self.finite_src_file = str(
QtGui.QFileDialog.getOpenFileName(
self,
"Choose *.srf or *.param File",
pwd,
"Standard Rupture Format (*.srf);;"
"USGS finite source files (*.param)",
))
if not self.finite_src_file:
return
if self.finite_src_file.endswith(".srf"):
self.finite_source = FiniteSource.from_srf_file(
self.finite_src_file, normalize=True)
elif self.finite_src_file.endswith(".param"):
self.finite_source = FiniteSource.from_usgs_param_file(
self.finite_src_file)
else:
raise IOError("unknown file type *.%s" %
self.finite_src_file.split(".")[-1])
self._setup_finite_source()
self.update()
self.set_info()
def on_open_srf_file_button_released(self):
pwd = os.getcwd()
self.finite_src_file = str(QtGui.QFileDialog.getOpenFileName(
self, "Choose *.srf or *.param File", pwd,
"Standard Rupture Format (*.srf);;"
"USGS finite source files (*.param)"))
if not self.finite_src_file:
return
if self.finite_src_file.endswith('.srf'):
self.finite_source = FiniteSource.from_srf_file(
self.finite_src_file, normalize=True)
elif self.finite_src_file.endswith('.param'):
self.finite_source = FiniteSource.from_usgs_param_file(
self.finite_src_file)
else:
raise IOError('unknown file type *.%s' %
self.finite_src_file.split('.')[-1])
self._setup_finite_source()
self.update()
self.set_info()
