Updated WaterFlow test

It's slightly more informative now and tests a bit more of WaterFlow's
functionality