[test] Add tests

Fixes #1.
Merge pull request #7 from mmalecki/tests

[test] Add tests