improve unit test
Merge pull request #53 from box/utest

improve unit test