def test1(self):
truncation_level = 3.4
imts = {imt.PGA(): [1, 2, 3], imt.PGD(): [2, 4]}
time_span = 49.2
rup11 = self.FakeRupture(0.23, const.TRT.ACTIVE_SHALLOW_CRUST)
rup12 = self.FakeRupture(0.15, const.TRT.ACTIVE_SHALLOW_CRUST)
rup21 = self.FakeRupture(0.04, const.TRT.VOLCANIC)
source1 = self.FakeSource([rup11, rup12], time_span=time_span)
source2 = self.FakeSource([rup21], time_span=time_span)
sources = iter([source1, source2])
site1 = self.FakeSite(Point(10, 20))
site2 = self.FakeSite(Point(20, 30))
sites = [site1, site2]
gsim1 = self.FakeGSIM(truncation_level, imts, poes={
(site1, rup11, imt.PGA()): [0.1, 0.05, 0.03],
(site2, rup11, imt.PGA()): [0.11, 0.051, 0.034],
(site1, rup12, imt.PGA()): [0.12, 0.052, 0.035],
(site2, rup12, imt.PGA()): [0.13, 0.053, 0.036],
(site1, rup11, imt.PGD()): [0.4, 0.33],
(site2, rup11, imt.PGD()): [0.39, 0.331],
(site1, rup12, imt.PGD()): [0.38, 0.332],
(site2, rup12, imt.PGD()): [0.37, 0.333],
})
gsim2 = self.FakeGSIM(truncation_level, imts, poes={
(site1, rup21, imt.PGA()): [0.5, 0.3, 0.2],
(site2, rup21, imt.PGA()): [0.4, 0.2, 0.1],
(site1, rup21, imt.PGD()): [0.24, 0.08],
(site2, rup21, imt.PGD()): [0.14, 0.09],
})
gsims = {const.TRT.ACTIVE_SHALLOW_CRUST: gsim1,
const.TRT.VOLCANIC: gsim2}
site1_pga_poe_expected = [0.0639157, 0.03320212, 0.02145989]
site2_pga_poe_expected = [0.06406232, 0.02965879, 0.01864331]
site1_pgd_poe_expected = [0.16146619, 0.1336553]
site2_pgd_poe_expected = [0.15445961, 0.13437589]
curves = hazard_curves_poissonian(sources, sites, imts, time_span,
gsims, truncation_level)
self.assertIsInstance(curves, dict)
self.assertEqual(set(curves.keys()), set([imt.PGA(), imt.PGD()]))
pga_curves = curves[imt.PGA()]
self.assertIsInstance(pga_curves, numpy.ndarray)
self.assertEqual(pga_curves.shape, (2, 3)) # two sites, three IMLs
site1_pga_poe, site2_pga_poe = pga_curves
self.assertTrue(numpy.allclose(site1_pga_poe, site1_pga_poe_expected),
str(site1_pga_poe))
self.assertTrue(numpy.allclose(site2_pga_poe, site2_pga_poe_expected),
str(site2_pga_poe))
pgd_curves = curves[imt.PGD()]
self.assertIsInstance(pgd_curves, numpy.ndarray)
self.assertEqual(pgd_curves.shape, (2, 2)) # two sites, two IMLs
site1_pgd_poe, site2_pgd_poe = pgd_curves
self.assertTrue(numpy.allclose(site1_pgd_poe, site1_pgd_poe_expected),
str(site1_pgd_poe))
self.assertTrue(numpy.allclose(site2_pgd_poe, site2_pgd_poe_expected),
str(site2_pgd_poe))
def test_point_sources(self):
sources = [
nhlib.source.PointSource(source_id='point1', name='point1',
tectonic_region_type=const.TRT.ACTIVE_SHALLOW_CRUST,
mfd=nhlib.mfd.EvenlyDiscretizedMFD(min_mag=4, bin_width=1,
occurrence_rates=[5]),
nodal_plane_distribution=nhlib.pmf.PMF([
(1, nhlib.geo.NodalPlane(strike=0.0, dip=90.0, rake=0.0))
]),
hypocenter_distribution=nhlib.pmf.PMF([(1, 10)]),
upper_seismogenic_depth=0.0,
lower_seismogenic_depth=10.0,
magnitude_scaling_relationship = nhlib.scalerel.PeerMSR(),
rupture_aspect_ratio=2,
rupture_mesh_spacing=1.0,
location=Point(10, 10)
),
nhlib.source.PointSource(source_id='point2', name='point2',
tectonic_region_type=const.TRT.ACTIVE_SHALLOW_CRUST,
mfd=nhlib.mfd.EvenlyDiscretizedMFD(min_mag=4, bin_width=2,
occurrence_rates=[5, 6, 7]),
nodal_plane_distribution=nhlib.pmf.PMF([
(1, nhlib.geo.NodalPlane(strike=0, dip=90, rake=0.0)),
]),
hypocenter_distribution=nhlib.pmf.PMF([(1, 10)]),
upper_seismogenic_depth=0.0,
lower_seismogenic_depth=10.0,
magnitude_scaling_relationship = nhlib.scalerel.PeerMSR(),
rupture_aspect_ratio=2,
rupture_mesh_spacing=1.0,
location=Point(10, 11)
),
]
sites = [nhlib.site.Site(Point(11, 10), 1, True, 2, 3),
nhlib.site.Site(Point(10, 16), 2, True, 2, 3),
nhlib.site.Site(Point(10, 10.6), 3, True, 2, 3),
nhlib.site.Site(Point(10, 10.7), 4, True, 2, 3)]
sitecol = nhlib.site.SiteCollection(sites)
from nhlib.gsim.sadigh_1997 import SadighEtAl1997
gsims = {const.TRT.ACTIVE_SHALLOW_CRUST: SadighEtAl1997()}
truncation_level = 1
time_span = 1.0
imts = {nhlib.imt.PGA(): [0.1, 0.5, 1.3]}
from nhlib.calc import filters
source_site_filter = self.SitesCounterSourceFilter(
filters.source_site_distance_filter(30)
)
rupture_site_filter = self.SitesCounterRuptureFilter(
filters.rupture_site_distance_filter(30)
)
hazard_curves_poissonian(
iter(sources), sitecol, imts, time_span, gsims, truncation_level,
source_site_filter=source_site_filter,
rupture_site_filter=rupture_site_filter
)
# there are two sources and four sites. first source should
# be filtered completely since it is too far from all the sites.
# the second one should take only three sites -- all except (10, 16).
# it generates three ruptures with magnitudes 4, 6 and 8, from which
# the first one doesn't affect any of sites and should be ignored,
# second only affects site (10, 10.7) and the last one affects all
# three.
self.assertEqual(source_site_filter.counts,
[('point2', [1, 3, 4])])
self.assertEqual(rupture_site_filter.counts,
[(6, [4]), (8, [1, 3, 4])])
