Add initial unit tests, runnable with py.test