Coverage for kwave/kWaveSimulation_helper/retract_transducer_grid_size.py: 27%
11 statements
« prev ^ index » next coverage.py v6.5.0, created at 2022-10-24 11:55 -0700
« prev ^ index » next coverage.py v6.5.0, created at 2022-10-24 11:55 -0700
1from kwave import NotATransducer
2import numpy as np
5def retract_transducer_grid_size(source, sensor, retract_size, pml_inside: bool):
6 # resize the transducer object if the grid has been expanded
7 is_source_kwave_transducer = isinstance(source, NotATransducer)
8 is_sensor_kwave_transducer = isinstance(sensor, NotATransducer)
9 retract_size = np.array(retract_size)
11 if not pml_inside and (is_source_kwave_transducer or is_sensor_kwave_transducer):
13 # check if the sensor is a transducer
14 if is_sensor_kwave_transducer:
15 # retract the transducer mask
16 sensor.retract_grid(retract_size)
18 # check if the source is a transducer, and if so, and different
19 # transducer to the sensor
20 if is_source_kwave_transducer and not (is_sensor_kwave_transducer and sensor == source):
22 # retract the transducer mask
23 source.retract_grid(retract_size)