Working through python tdd book