Added test cases to FloodGeneratorTest.

Also moved unit test classes to the correct packages with respect to the classes they
are testing (this was no longer the case after moving some classes to the 'utility' package).