Add unit tests for WavePropagation