add unit test codes