Coverage for src/flexfrac1d/lib/_ph_utils.py: 100%

1import numpy as np 

2 

3 

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

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

6 

7 

8def _wavefield( 

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))