Following up examples in TDD with python book