如何使用输入对此函数进行单元测试