Py.test Example Code