Add unit tests

[Mokto]

I need to add some units tests before final release.

Not much, but it would avoid some issues.

[fakenickels]

Necessary indeed so we can lower the chances of regressions.