Write proper unit tests

We use benchmark.py for tests, but having a proper unit test framework is
a very good idea. Hence write a framework / working stub for unit tests,
to be used together with benchmark.py.