[test] Add tests

Fixes #1.
Merge remote branch 'mmalecki/tests'