为上述代码写一个简单的测试模块

可以参考以下测试模块的代码：

import unittest
from chitgpt import ChitGPT

class TestChitGPT(unittest.TestCase):

    def setUp(self):
        self.chitgpt = ChitGPT()

    def test_response(self):
        user_input = "你好"
        expected_response = "你好，请问有什么需要帮助的吗？"
        response = self.chitgpt.response(user_input)
        self.assertEqual(response, expected_response)

if __name__ == '__main__':
    unittest.main()

在上述测试模块中，我们首先导入了unittest模块和我们自己编写的ChitGPT类。然后，在setUp()方法中创建了一个ChitGPT对象用于测试。接下来，我们定义了一个test_response()方法，用于测试ChitGPT类的response()方法是否能够正确地响应用户输入。在这个方法中，我们传入了一个用户输入“你好”，并期望得到的回复是“你好，请问有什么需要帮助的吗？”。最后，我们使用assertEqual()方法进行断言，判断实际得到的回复是否与期望的一致。如果测试通过，就可以运行unittest.main()方法来执行测试。