add unit test

add unit test