修改测试类

Signed-off-by: capricemz <522382667@qq.com>