Improve the tests by using a helper function for modifying the input file and removed the redundant code