exercise in unit testing python