adding a draft of test_shallow_water