Coverage for src/swiift/lib/_ph_utils.py: 60%

1import numpy as np (empty)

2 

3 

4def _unit_wavefield(x: np.ndarray, c_wavenumbers: np.ndarray) -> np.ndarray: (empty)

5 return np.exp((1j * c_wavenumbers[:, None]) * x) 

6 

7 

8def _wavefield( (empty)

9 x: np.ndarray, c_amps: np.ndarray, c_wavenumbers: np.ndarray 

10) -> np.ndarray: 

11 return np.imag(c_amps @ _unit_wavefield(x, c_wavenumbers))