Write some basic tests for core logic

We need some unit tests to ensure that the base functionality will not break with time.
Possible solution is a pytest based test set which can be placed under a test directory (separated from the source #12 will reorganize the source anyway).