Add unit test

[peter279k]

To let the package behavior be reliable, I think we should consider adding the unit tests.

Once this issue is accepted, I can work on this.