Add unit tests

[morcuended]

With unit test is easier for a new developer to not break the code with new changes and we are sure that the code does what it is supposed to do.

https://docs.pytest.org/en/stable/example/simple.html

Developers should have pytest package installed in their environments.