添加单元测试功能