def test_hazard_curve(self):
output = self.FakeOutput('hazard_curve')
curves = [
self.FakeHazardCurve(output, self.ltr_mc, 'PGA', None, None, None),
self.FakeHazardCurve(output, self.ltr_enum, 'SA', 0.025, None,
None),
self.FakeHazardCurve(output, None, 'SA', 0.025, 'mean', None),
self.FakeHazardCurve(output, None, 'SA', 0.025, 'quantile', 0.85),
]
expected_paths = [
'%s/calc_7/hazard_curve/PGA/'
'hazard_curve-smltp_B1_B3-gsimltp_B2_B4-ltr_3.xml',
'%s/calc_7/hazard_curve/SA-0.025/'
'hazard_curve-smltp_B10_B9-gsimltp_B7_B8.xml',
'%s/calc_7/hazard_curve/SA-0.025/'
'hazard_curve-mean.xml',
'%s/calc_7/hazard_curve/SA-0.025/'
'hazard_curve-quantile_0.85.xml',
]
expected_paths = [x % self.target_dir for x in expected_paths]
for i, curve in enumerate(curves):
self.assertEqual(
expected_paths[i],
hazard._get_result_export_path(7, self.target_dir, curve)
)
def test_hazard_map(self):
output = self.FakeOutput('hazard_map')
maps = [
self.FakeHazardMap(output, self.ltr_mc, 'PGA', None, 0.1,
None, None),
self.FakeHazardMap(output, self.ltr_mc, 'SA', 0.025, 0.2,
None, None),
self.FakeHazardMap(output, None, 'SA', 0.025, 0.3,
'mean', None),
self.FakeHazardMap(output, None, 'SA', 0.025, 0.4,
'quantile', 0.85),
]
expected_paths = [
'%s/calc_7/hazard_map/PGA/'
'hazard_map-poe_0.1-smltp_B1_B3-gsimltp_B2_B4-ltr_3.xml',
'%s/calc_7/hazard_map/SA-0.025/'
'hazard_map-poe_0.2-smltp_B1_B3-gsimltp_B2_B4-ltr_3.xml',
'%s/calc_7/hazard_map/SA-0.025/'
'hazard_map-poe_0.3-mean.xml',
'%s/calc_7/hazard_map/SA-0.025/'
'hazard_map-poe_0.4-quantile_0.85.xml',
]
expected_paths = [x % self.target_dir for x in expected_paths]
for i, hmap in enumerate(maps):
self.assertEqual(
expected_paths[i],
hazard._get_result_export_path(7, self.target_dir, hmap)
)
def test_clt_ses(self):
output = self.FakeOutput('complete_lt_ses')
ses = self.FakeCLTSES(output)
expected_path = '%s/calc_10/ses/complete_lt_ses.xml'
expected_path %= self.target_dir
self.assertEqual(
expected_path,
hazard._get_result_export_path(10, self.target_dir, ses)
)
def test_clt_gmf(self):
output = self.FakeOutput('complete_lt_gmf')
gmf = self.FakeCLTGMF(output)
expected_path = '%s/calc_9/gmf/complete_lt_gmf.xml'
expected_path %= self.target_dir
self.assertEqual(
expected_path,
hazard._get_result_export_path(9, self.target_dir, gmf)
)
def test_ses(self):
output = self.FakeOutput('ses')
ses = self.FakeGMF(output, self.ltr_mc)
expected_path = (
'%s/calc_8/ses/ses-smltp_B1_B3-gsimltp_B2_B4-ltr_3.xml'
% self.target_dir
)
self.assertEqual(
expected_path,
hazard._get_result_export_path(8, self.target_dir, ses)
)
def test_gmf(self):
output = self.FakeOutput('gmf')
gmf = self.FakeGMF(output, self.ltr_enum)
expected_path = (
'%s/calc_8/gmf/gmf-smltp_B10_B9-gsimltp_B7_B8.xml'
% self.target_dir
)
self.assertEqual(
expected_path,
hazard._get_result_export_path(8, self.target_dir, gmf)
)
def test_uhs(self):
output = self.FakeOutput('uh_spectra')
uh_spectra = [
self.FakeUHS(output, self.ltr_mc, 0.1, None, None),
self.FakeUHS(output, None, 0.2, 'mean', None),
self.FakeUHS(output, None, 0.3, 'quantile', 0.85),
]
expected_paths = [
'%s/calc_7/uh_spectra/'
'uh_spectra-poe_0.1-smltp_B1_B3-gsimltp_B2_B4-ltr_3.xml',
'%s/calc_7/uh_spectra/uh_spectra-poe_0.2-mean.xml',
'%s/calc_7/uh_spectra/uh_spectra-poe_0.3-quantile_0.85.xml',
]
expected_paths = [x % self.target_dir for x in expected_paths]
for i, uhs in enumerate(uh_spectra):
self.assertEqual(
expected_paths[i],
hazard._get_result_export_path(7, self.target_dir, uhs)
)
def test_disagg(self):
output = self.FakeOutput('disagg_matrix')
matrices = [
self.FakeDisagg(output, self.ltr_mc, 'PGA', None,
self.Location(33.333, -89.999001)),
self.FakeDisagg(output, self.ltr_enum, 'SA', 0.025,
self.Location(40.1, 10.1)),
]
expected_paths = [
'%s/calc_7/disagg_matrix/PGA/'
'disagg_matrix-lon_33.333-lat_-89.999001-smltp_B1_B3-'
'gsimltp_B2_B4-ltr_3.xml',
'%s/calc_7/disagg_matrix/SA-0.025/'
'disagg_matrix-lon_40.1-lat_10.1-smltp_B10_B9-gsimltp_B7_B8.xml'
]
expected_paths = [x % self.target_dir for x in expected_paths]
for i, matrix in enumerate(matrices):
self.assertEqual(
expected_paths[i],
hazard._get_result_export_path(7, self.target_dir, matrix)
)
