def test_assign_receiver_points():
rec_counts, _ = ww.calculate_receiver_window_counts(windows)
def _assert(_points):
for p in _points:
if p.tag == "II.AAK..BHZ":
npt.assert_almost_equal(p.latitude, 0.0)
npt.assert_almost_equal(p.longitude, 0.0)
elif p.tag == "II.ABKT..BHZ":
npt.assert_almost_equal(p.latitude, 0.0)
npt.assert_almost_equal(p.longitude, 120.0)
elif p.tag == "IU.BCD..BHZ":
npt.assert_almost_equal(p.latitude, 0.0)
npt.assert_almost_equal(p.longitude, -120.0)
points = ww.assign_receiver_to_points(rec_counts["BHZ"], stations)
assert len(points) == 3
_assert(points)
points = ww.assign_receiver_to_points(rec_counts["BHR"], stations)
assert len(points) == 3
_assert(points)
points = ww.assign_receiver_to_points(rec_counts["BHT"], stations)
assert len(points) == 2
_assert(points)
def test_assign_receiver_points():
rec_counts, _ = ww.calculate_receiver_window_counts(windows)
def _assert(_points):
for p in _points:
if p.tag == "II.AAK..BHZ":
npt.assert_almost_equal(p.latitude, 0.0)
npt.assert_almost_equal(p.longitude, 0.0)
elif p.tag == "II.ABKT..BHZ":
npt.assert_almost_equal(p.latitude, 0.0)
npt.assert_almost_equal(p.longitude, 120.0)
elif p.tag == "IU.BCD..BHZ":
npt.assert_almost_equal(p.latitude, 0.0)
npt.assert_almost_equal(p.longitude, -120.0)
points = ww.assign_receiver_to_points(rec_counts["BHZ"], stations)
assert len(points) == 3
_assert(points)
points = ww.assign_receiver_to_points(rec_counts["BHR"], stations)
assert len(points) == 3
_assert(points)
points = ww.assign_receiver_to_points(rec_counts["BHT"], stations)
assert len(points) == 2
_assert(points)
def test_get_receiver_weights():
center = SpherePoint(0, 0, tag="source")
rec_counts, _ = ww.calculate_receiver_window_counts(windows)
points = ww.assign_receiver_to_points(rec_counts["BHZ"], stations)
ref_distance, cond_number = ww.get_receiver_weights(
"BHZ", center, points, 0.35, plot=False)
for p in points:
npt.assert_almost_equal(p.weight, 1.0)
npt.assert_almost_equal(cond_number, 1.0)
points = ww.assign_receiver_to_points(rec_counts["BHT"], stations)
ref_distance, cond_number = ww.get_receiver_weights(
"BHZ", center, points, 0.35, plot=False)
for p in points:
npt.assert_almost_equal(p.weight, 1.0)
npt.assert_almost_equal(cond_number, 1.0)
def test_get_receiver_weights():
center = SpherePoint(0, 0, tag="source")
rec_counts, _ = ww.calculate_receiver_window_counts(windows)
points = ww.assign_receiver_to_points(rec_counts["BHZ"], stations)
ref_distance, cond_number = ww.get_receiver_weights(
"BHZ", center, points, 0.35, plot=False)
for p in points:
npt.assert_almost_equal(p.weight, 1.0)
npt.assert_almost_equal(cond_number, 1.0)
points = ww.assign_receiver_to_points(rec_counts["BHT"], stations)
ref_distance, cond_number = ww.get_receiver_weights(
"BHZ", center, points, 0.35, plot=False)
for p in points:
npt.assert_almost_equal(p.weight, 1.0)
npt.assert_almost_equal(cond_number, 1.0)
def test_normalize_receiver_weights():
rec_counts, cat_wcounts = ww.calculate_receiver_window_counts(windows)
comp = "BHZ"
channels = list(rec_counts[comp].keys())
channels.sort()
points = ww.assign_receiver_to_points(channels, stations)
weights = ww.normalize_receiver_weights(points, rec_counts[comp])
assert len(weights) == 3
for v in weights.values():
npt.assert_almost_equal(v, 1.0)
points[0].weight = 0.5
points[1].weight = 0.75
points[2].weight = 1.0
weights = ww.normalize_receiver_weights(points, rec_counts[comp])
assert len(weights) == 3
npt.assert_almost_equal(weights["II.AAK..BHZ"], 0.625)
npt.assert_almost_equal(weights["II.ABKT..BHZ"], 0.9375)
npt.assert_almost_equal(weights["IU.BCD..BHZ"], 1.25)
def test_normalize_receiver_weights():
rec_counts, cat_wcounts = ww.calculate_receiver_window_counts(windows)
comp = "BHZ"
channels = rec_counts[comp].keys()
channels.sort()
points = ww.assign_receiver_to_points(channels, stations)
weights = ww.normalize_receiver_weights(points, rec_counts[comp])
assert len(weights) == 3
for v in weights.itervalues():
npt.assert_almost_equal(v, 1.0)
points[0].weight = 0.5
points[1].weight = 0.75
points[2].weight = 1.0
weights = ww.normalize_receiver_weights(points, rec_counts[comp])
assert len(weights) == 3
npt.assert_almost_equal(weights["II.AAK..BHZ"], 0.625)
npt.assert_almost_equal(weights["II.ABKT..BHZ"], 0.9375)
npt.assert_almost_equal(weights["IU.BCD..BHZ"], 1.25)
