测试文本生成功能

input_text = "Tell me a joke."
expected_output_text = "Why don't scientists trust atoms? Because they make up everything!"
generated_text = mock_text_generation(input_text)
assert generated_text == expected_output_text, f"Text generation test failed: expected {expected_output_text}, got {generated_text}"
print("Text generation test passed")
Text generation test passed

测试代码生成功能

input_code = "def greet(name):"
expected_output_code = "def greet(name):\n return f'Hello, {name}!'"
generated_code = mock_code_generation(input_code)
assert generated_code == expected_output_code, f"Code generation test failed: expected {expected_output_code}, got {generated_code}"
print("Code generation test passed")
Code generation test passed