cleanup test and levee test

Test for levee update function