Added a test for the file generator