Add unit tests

[elichad]

Recommend using pytest for this as you can use the https://pypi.org/project/pytest-cov/ plugin to check test coverage