update unit test
Merge pull request #7 from anhtuannguyentran/v2

update unit test