Basic boilerplate for doctests in python