add some unit test code

add some unit test code