Following "TDD with python" book