adding tests for generate