Set up basic testing for generator