Added new test for generate_output method.