unit test

add unit test