def test_read_gmfs(self):
"""Verify _get_db_gmfs."""
params = {
'REGION_VERTEX': '40,-117, 42,-117, 42,-116, 40,-116',
'REGION_GRID_SPACING': '1.0'}
the_job = helpers.create_job(params, job_id=self.job.id)
calculator = EventBasedRiskCalculator(the_job)
self.assertEqual(3, len(calculator._gmf_db_list(self.job.id)))
# only the keys in gmfs are used
gmfs = calculator._get_db_gmfs([], self.job.id)
self.assertEqual({}, gmfs)
# only the keys in gmfs are used
sites = [Site(lon, lat)
for lon in xrange(-117, -115)
for lat in xrange(40, 43)]
gmfs = calculator._get_db_gmfs(sites, self.job.id)
# avoid rounding errors
for k, v in gmfs.items():
gmfs[k] = [round(i, 1) for i in v]
self.assertEqual({
'0!0': [0.1, 0.5, 0.0],
'0!1': [0.2, 0.6, 0.0],
'1!0': [0.4, 0.8, 1.3],
'1!1': [0.3, 0.7, 1.2],
'2!0': [0.0, 0.0, 1.0],
'2!1': [0.0, 0.0, 1.1],
}, gmfs)
def test_read_gmfs(self):
"""Verify _get_db_gmfs."""
params = {
'REGION_VERTEX': '40,-117, 42,-117, 42,-116, 40,-116',
'REGION_GRID_SPACING': '1.0'
}
the_job = helpers.create_job(params, job_id=self.job.id)
calculator = EventBasedRiskCalculator(the_job)
self.assertEqual(3, len(calculator._gmf_db_list(self.job.id)))
# only the keys in gmfs are used
gmfs = calculator._get_db_gmfs([], self.job.id)
self.assertEqual({}, gmfs)
# only the keys in gmfs are used
sites = [
Site(lon, lat) for lon in xrange(-117, -115)
for lat in xrange(40, 43)
]
gmfs = calculator._get_db_gmfs(sites, self.job.id)
# avoid rounding errors
for k, v in gmfs.items():
gmfs[k] = [round(i, 1) for i in v]
self.assertEqual(
{
'0!0': [0.1, 0.5, 0.0],
'0!1': [0.2, 0.6, 0.0],
'1!0': [0.4, 0.8, 1.3],
'1!1': [0.3, 0.7, 1.2],
'2!0': [0.0, 0.0, 1.0],
'2!1': [0.0, 0.0, 1.1],
}, gmfs)
