[AWSReInventPartnerSessions2024] Simulate Amazon Q query for code generation
generated_code = f"def process_data(data):\n return sorted(data)\n"
return generated_code
def test_code(code, test_cases):
exec(code)
for case in test_cases:
input_data, expected = case
result = process_data(input_data)
if result != expected:
return False
return True