Add unit and integration tests

[mpaluchowski]

The package is missing any tests for the moment. We need to add some.