def test_ray_weights_for_views():
context = make_context()
views = context["views"]
paths = context["paths"]
paths_set = set(paths.values())
ray_weights_cache = bim.ray_weights_for_views(
views,
frequency=context["freq"],
probe_element_width=context["element_width"])
assert ray_weights_cache.tx_ray_weights_debug_dict is None
assert ray_weights_cache.rx_ray_weights_debug_dict is None
assert len(paths) >= len(ray_weights_cache.tx_ray_weights_dict) > 3
assert len(paths) >= len(ray_weights_cache.rx_ray_weights_dict) > 3
assert set(ray_weights_cache.rx_ray_weights_dict.keys()) == paths_set
nbytes_without_debug = ray_weights_cache.nbytes
assert nbytes_without_debug > 0
ray_weights_cache = bim.ray_weights_for_views(
views,
frequency=context["freq"],
probe_element_width=context["element_width"],
save_debug=True,
)
assert (ray_weights_cache.tx_ray_weights_debug_dict.keys() ==
ray_weights_cache.tx_ray_weights_dict.keys())
assert (ray_weights_cache.rx_ray_weights_debug_dict.keys() ==
ray_weights_cache.rx_ray_weights_dict.keys())
assert len(paths) >= len(ray_weights_cache.tx_ray_weights_dict) > 3
assert len(paths) >= len(ray_weights_cache.rx_ray_weights_dict) > 3
assert set(ray_weights_cache.rx_ray_weights_dict.keys()) == paths_set
nbytes_with_debug = ray_weights_cache.nbytes
assert nbytes_with_debug > nbytes_without_debug
def test_make_views():
context = make_context()
probe_oriented_points = context["probe_oriented_points"]
scatterer_oriented_points = context["scatterer_oriented_points"]
exam_obj = context["exam_obj"]
views = bim.make_views(exam_obj, probe_oriented_points,
scatterer_oriented_points)
assert list(views.keys()) == list(context["views"].keys())
def test_ray_weights():
context = make_context()
paths = context["paths"]
""":type : dict[str, arim.Path]"""
ray_geometry_dict = context["ray_geometry_dict"]
""":type : dict[str, arim.path.RayGeometry]"""
model_options = dict(
frequency=context["freq"],
probe_element_width=context["element_width"],
use_beamspread=True,
use_directivity=True,
use_transrefl=True,
use_attenuation=True,
)
for pathname, path in paths.items():
# Direct
ray_geometry = ray_geometry_dict[pathname]
ray_geometry2 = arim.ray.RayGeometry.from_path(path, use_cache=False)
weights, weights_dict = bim.tx_ray_weights(path, ray_geometry,
**model_options)
weights2, _ = bim.tx_ray_weights(path, ray_geometry2, **model_options)
assert "beamspread" in weights_dict
assert "directivity" in weights_dict
assert "transrefl" in weights_dict
assert "attenuation" in weights_dict
np.testing.assert_allclose(weights, weights2)
# Reverse
ray_geometry = ray_geometry_dict[pathname]
ray_geometry2 = arim.ray.RayGeometry.from_path(path, use_cache=False)
weights, weights_dict = bim.rx_ray_weights(path, ray_geometry,
**model_options)
weights2, _ = bim.rx_ray_weights(path, ray_geometry2, **model_options)
assert "beamspread" in weights_dict
assert "directivity" in weights_dict
assert "transrefl" in weights_dict
assert "attenuation" in weights_dict
np.testing.assert_allclose(weights, weights2)