Add tests

[nickescobedo]

I created some tests for your package. Although it doesn't cover all, it's a good start. Hope this helps. 😄

If you have any questions let me know!