Writing unit tests with Python unittest