add unit tests

[klaeufer]

https://docs.pytest.org/en/stable/goodpractices.html