Add unit test
Merge pull request #1 from wilsonge/patch-19

Add unit test