add test code

Signed-off-by: yuanhuan2005 <yuanhuan_2005@126.com>