Add tests

[TimothyGu]

Fixes #2.

There's still room for improvements, especially for the error handling part (I'd like to wait on this until #6 is fixed).

=============================== Coverage summary ===============================
Statements   : 95.7% ( 89/93 )
Branches     : 92.06% ( 58/63 )
Functions    : 94.12% ( 16/17 )
Lines        : 95.56% ( 86/90 )
================================================================================

After #5:

=============================== Coverage summary ===============================
Statements   : 97.85% ( 91/93 )
Branches     : 95.24% ( 60/63 )
Functions    : 100% ( 17/17 )
Lines        : 97.75% ( 87/89 )
================================================================================