example from "TDD with Python" book