Added basic unit test cases.