correct unit test cases
Merge pull request #4 from bobbyc/dev

correct unit test cases