working through TDD Python Book