add unit tests

[tgoelles]

Interesting project, but please consider adding automatic tests with pytest https://docs.pytest.org/en/7.1.x/
For maintainability and increased trust in the code.