Add proper tests

[driade]

The current test it's very basic.

[driade]

Still open to improvements if someone wants to check @jhoughtelin comments at #5