修改为main方法测试

Signed-off-by: stphenZkang <qiao54007@gmail.com>