Add some unit test codes