添加、优化测试类