Learning python with unit test