Working through o'reilly's tdd python book