Add unit test

[Allerito]

It would be nice for the project to have tests which check that the main functions are working properly